https://atcoder.jp/contests/agc023/tasks/agc023_a
\(C_j = \sum_{k=1}^j A_k\) とおくと、 \(\sum_{k=i}^j A_k = 0 \Leftrightarrow C_j = C_{k-1}\) であるから、 累積和を求めて \(C_0=0, C_j = \sum_{k=1}^j A_k\) とおき、 \(I_x = \{ i \mid C_i = x\}\) を計算して、\(\sum_x |I_x| (|I_x|-1)/2\) を求めれば良い。