https://atcoder.jp/contests/agc023/tasks/agc023_a Cj=∑k=1jAkC_j = \sum_{k=1}^j A_kCj=∑k=1jAk とおくと、 ∑k=ijAk=0⇔Cj=Ck−1\sum_{k=i}^j A_k = 0 \Leftrightarrow C_j = C_{k-1}∑k=ijAk=0⇔Cj=Ck−1 であるから、 累積和を求めて C0=0,Cj=∑k=1jAkC_0=0, C_j = \sum_{k=1}^j A_kC0=0,Cj=∑k=1jAk とおき、 Ix={i∣Ci=x}I_x = \{ i \mid C_i = x\}Ix={i∣Ci=x} を計算して、∑x∣Ix∣(∣Ix∣−1)/2\sum_x |I_x| (|I_x|-1)/2∑x∣Ix∣(∣Ix∣−1)/2 を求めれば良い。 https://atcoder.jp/contests/agc023/submissions/32455933