Codeforces Round #632 (Div. 2) C. Eugene and an array

整理的算法模板：ACM算法模板总结（分类详细版）

题目链接：http://codeforces.com/contest/1333/problem/C

 

题意：

• 子数组：数组a的任意一个连续的子序列，称作数组a的子数组；
• 好数组：数组b的任意子数组的和都不为0

求出一个数组的所有好数组；

思路：

（昨天晚上准备开c题的时候，家里出事了，于是乎今天在医院写下这篇题解；）

首先暴力绝对gg，当时看这道题第一感觉就是前缀和+map，但是还没想起来怎么做就走了；

具体思路就是固定当前右端点，看左端点有多少种取法；

因为如果前缀和区间一旦存在某个Sum[i] == Sum[j]的话，则说明a[i+1]+a[i+2]....+a[j] = 0

在求前缀和的过程中，当前的右端点为i，那么前面的i-1个位置包括i，都可以作为左端点，但是前面如果存在和为0的区间，那么左端点只能从最靠近i的那个区间为0的左边算起；也就是好数组的个数为 i - max(L)；

 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
ll a[2*100005];
map<ll,ll> mp;

int main()
{
    int n;
    cin >>n;
    ll ans=0,q=-1;
    mp[0]=0;
    for(int i=1;i<=n;i++) 
    {
        cin >>a[i];
        a[i]+=a[i-1];
        if(mp.count(a[i])) q=max(q,mp[a[i]]);
        ans+=i-q-1;
        mp[a[i]]=i;
        
    }
    cout <<ans<<endl;
}