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);
}