https://atcoder.jp/contests/agc023/tasks/agc023_a

Cj=k=1jAkC_j = \sum_{k=1}^j A_k とおくと、 k=ijAk=0Cj=Ck1\sum_{k=i}^j A_k = 0 \Leftrightarrow C_j = C_{k-1} であるから、 累積和を求めて C0=0,Cj=k=1jAkC_0=0, C_j = \sum_{k=1}^j A_k とおき、 Ix={iCi=x}I_x = \{ i \mid C_i = x\} を計算して、xIx(Ix1)/2\sum_x |I_x| (|I_x|-1)/2 を求めれば良い。

https://atcoder.jp/contests/agc023/submissions/32455933