Codeforces Round #632 (Div. 2) C. Eugene and an array (map&前缀和)

题目传送门

题意:求给定数组所有区间和非0的区间个数。

思路:



AC代码:

#include<iostream>
#include<map>
using namespace std;
typedef long long ll;
map<ll,ll>mp;
ll n,sum,mx,ans;//mx满足(前一个区间)[idx,i-1]==0和[idx,i]==0的最大idx 
int main(){
	mp[0]=1;//初始化保证第一个前缀和[1,i]为0的右端i的非0区间个数为i-1 
	cin>>n;
	for(int i=1,x;i<=n;i++){
		cin>>x,sum+=x;
		mx=max(mx,mp[sum]);
		ans+=i-mx;
		mp[sum]=i+1;
	}
	cout<<ans<<endl;
	return 0;
}