CF1333 C. Eugene and an array

C. Eugene and an array

题意

给你一个长度为n的数组，求不含和为0的子串的个数。

思路

前缀和 思维
前缀和$pre[i]=pre[j]$意味着$a_{i+1}\sim a_j$这一段的和为0，则所求字符串不应该包含这段，令$p$为$pre[i]$所在的位置，则不包含子串和为0的个数为$i-(p+1)$。
初始化$pos[0]=0$。

map.count(key) 返回值为bool型，表示是否存在key。

#include
using namespace std;
typedef long long ll;

ll n,q,s,ans;
map<ll,ll> pos;

int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cin>>n;
	pos[0]=0;
	ll q=-1;
	for(int i=1;i<=n;i++){
		ll x;cin>>x;
		s+=x;
		if(pos.count(s)) q=max(q,pos[s]);
		ans+=i-q-1;
		pos[s]=i;
	}
	cout<<ans<<'\n';
	return 0;
}