题意:
学长是个1,所以他非常不喜欢0,他的目标是消灭数组中所有的0, 有一个数组a,其中不存在a_i=0,但是丧心病狂的TA觉得这还不能够满足 即使是这个数组的任意子段的和为0也是不能接受的。 他可以往数组中的任意位置插入任意大小的数,请问最少插入几个数后才能成为的理想数组(即不存在子段和为0的情况)
解:
用set存,由于set有count函数,可以直接统计某个数字出现的次数
我们先将0 存入,再将每个数字前缀和存入,如果前缀和出现两次,代表我们必须插入一个大数字,或者0出现了一次,我们得插入一个大数字
例如:1 -5 2 3 2 -2 3
0 进1 进 -4进 -2进 1 进 1 出现两次,计数加一, 然后清空set,然后将0 与1 放入,1+2 进 1+2-2 = 1 进, 1进两次,计数加一,清空set 0, 1放入, 然后1+3 进 此时结束
计数为2
输出2
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6+6;
int n;
ll a[maxn],sum[maxn], sum1[maxn];
int main()
{
int n;
cin>>n;
for(int i = 1; i <= n; i ++)
cin>>a[i];
set<ll> s;
s.insert(0);
ll ans=0,res=0;
for(int i = 1; i <= n; i ++)
{
res += a[i];
if(s.count(res))
{
ans++;
s.clear();
s.insert(0);
res=a[i];
}
s.insert(res);
}
cout<<ans<<endl;
return 0;
}