题意:
给定a数组,询问有多少个区间之和不为0
解析:
逆向思维,我们找出有多少个区间之和为0得
思路 sum[i] 表示前缀和
如果sum[i]=sum[j] 那就说明 i+1~j这段区间之和一定是0.所以我们找到第一次出现的位置(假设为i) 那么贡献就是 i+1 。因为i前面所有区间都会包括这段为0得区间,+1是因为当前区间之和为0
位置用map记录
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=2e5+10000;
int a[N];
ll sum;
int n;
ll res,x;
map<ll ,int > v;
int main()
{
scanf("%d",&n);
v[0]=0;
int r=-1;
for(int i=1;i<=n;i++)
{
cin>>x;
res+=x;
if(v.count(res)) r=max(r,v[res]);
sum=sum+r+1;
v[res]=i;
}
cout<<(n+1ll)*n/2-sum<<endl;
}