题目传送门
题意: 对于一个连续子串来说,他没有总和为0的子串,我们就说这个子串是good,给你一个数组,求出good串的数量。
思路: 假设我们在a[i]处,以a[i]为结尾的子串就有i个,假设a[j]~a[k]的和为0,且j<k<=i,那么以a[i]为结尾的合法子串只有i-j个。我们处理一下前缀和,用map记录前缀和出现的位置,我们只要处理离i最近的那一个为0子串。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N=1e6+5;
const int inf=0x7fffffff;
const int mod=1e9+7;
const int eps=1e-6;
typedef long long ll;
typedef unsigned long long ull;
#define ls p<<1
#define rs p<<1|1
#define fi first
#define se second
#define mp make_pair
#define pb push_back
#define pii pair<int,int>
#define int long long
#define IOS ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define endl '\n'
#define null NULL
int a[N];
signed main()
{
IOS;
int n;
cin>>n;
map<int,int>ma;
int s=0;
ma[0]=0;
int q=-1;
int ans=0;
for(int i=1;i<=n;i++)
{
cin>>a[i];
s+=a[i];
if(ma[s]||s==0)
q=max(q,ma[s]);
ans+=i-q-1;
ma[s]=i;
}
cout<<ans<<endl;
}