D. Non-zero Segments

这题就是找前缀和相同的两个下标,因为sum[r]-sum[l-1]=0则满足[l,r]区间和为0。

直接正向遍历,每次找到一个相同的前缀和就清除之前做过的所有标记(这样能保证间隔最小),答案+1,同时重新计数前缀和。

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+10;
typedef long long ll;
unordered_map<ll,bool>vis;
int n,a[N];
ll sum[N];
int main()
{
    ios::sync_with_stdio(false);
    cin>>n;
    int ans=0;
    vis[0]=1;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        sum[i]=sum[i-1]+a[i];
        if(vis[sum[i]])
        {
            ans++;
            vis.clear(); // 复杂度O(n),与map中元素个数有关,最多累积清空n个
            sum[i]=a[i]; // 重新计数前缀和,此时第i项相当于下一次的第1项
            vis[0]=1; // 标记第0项
            vis[sum[i]]=1; // 标记第i项(相当于第1项)
        }
        else
        {
            vis[sum[i]]=1;
        }
    }
    printf("%d\n",ans);
    return 0;
}
/*
6
-2 1 -1 1 -1 2
ans:3

10
-2 2 1 1 1 -1 1 2 -2 -2
ans:4
*/