C. Eugene and an array(前缀和,map)

题目传送门

题意: 对于一个连续子串来说,他没有总和为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;

}

发布了93 篇原创文章 · 获赞 9 · 访问量 4203

猜你喜欢

转载自blog.csdn.net/Joker_He/article/details/105422454