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
*/