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$ を求めれば良い。