Codeforces Round #632 (Div. 2) C. Eugene and an array

Codeforces Round #632 (Div. 2) C. Eugene and an array

题目链接 ：https://codeforces.ml/contest/1333/problem/C
题意：给出一个长度为 n 的数组，求该数组所有好的子区间的数目。好的子区间的定义：该区间的所有子区间 的和不为0.
思路：
这题比赛的时候没写出来，知道怎么求和为0 的子区间，但求数量的时候没转过来，一直想用所有的区间减去非法区间。

可以用 map 记录前缀和 pre 出现的位置，如果 pre 出现了两次就说明该段的和为 0. 包含该段在内的区间是不可行的，所以答案 加上 index-map[pre]. index 表示以index为右端点的区间数量。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+7;
map<ll,ll>mp;
ll n,ans=0,pre=0,a[maxn],l;
int main (){
    cin>>n;mp[0]=0;
    for(int i=1;i<=n;i++){
        scanf("%lld",&a[i]);
        pre+=a[i];
        if(mp.count(pre)) l=max(l,mp[pre]+1);
        ans+=i-l;
        mp[pre]=i;
    }
    printf ("%lld\n",ans);

}