求O(n^2)代码

自己推了一个式子，感觉跟题解的 $O(n^2)$ 做法差不多（可能一样），可是自己打的代码总是只有25分，subtask 3，4 WA了，求大佬代码 QAQ。