题目
题意: 给定一个数组,可以把某个为0的数变为任意数。使数组的前缀和和为0尽可能地多。(s1、s2…sn)
思路: 贪心。先求出前缀和数组。用0把每个数组分隔开。第一个0与第二个0之间,把第一个0变为出现次数最多的数x的相反数-x。它并不影响下一个0影响的范围,假设第二个0与第三个0之间出现次数最多的数是y,y-x仍是对应区间出现次数最多的数。
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5+10;
int n,m,k,T;
int a[N];
ll s[N];
void solve()
{
cin>>n;
vector<int> va;
for(int i=1;i<=n;++i)
{
cin>>a[i];
s[i] = s[i-1] + a[i];
if(a[i]==0) va.push_back(i);
}
int ans = 0;
for(int i=0;i<va.size();++i)
{
map<ll,int> mp;
int mx = 0;
int r;
if(i<va.size()-1) r = va[i+1];
else r = n+1;
for(int j=va[i];j<r;++j)
{
mp[s[j]]++;
if(mp[s[j]]>mx) mx = mp[s[j]];
}
ans += mx;
}
if(va.size()>0)
{
for(int i=1;i<va[0];++i) if(s[i]==0) ans ++;
}
else
{
for(int i=1;i<=n;++i) if(s[i]==0) ans ++;
}
// cout<<s[n]<<"??\n";
// ll now = 0;
// for(int i=1;i<=n;++i)
// {
// cout<<a[i]<<" ";
// now += a[i];
// if(now==0) ans ++ ;
// }
// cout<<"\n";
cout<<ans<<"\n";
}
signed main(void)
{
ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
cin>>T;
while(T--) solve();
return 0;
}
/*
1
-2
1
-1
1
0
1
1
1
2
2
-2 -2
2
-2 -1
2
-2 0
2
-2 1
2
-2 2
2
-1 -2
2
-1 -1
2
-1 0
2
-1 1
2
-1 2
2
0 -2
2
0 -1
2
0 0
2
0 1
2
0 2
2
1 -2
2
1 -1
2
1 0
2
1 1
2
1 2
2
2 -2
2
2 -1
2
2 0
2
2 1
2
2 2
3
-2 -2 -2
3
-2 -2 -1
3
-2 -2 0
3
-2 -2 1
3
-2 -2 2
3
-2 -1 -2
3
-2 -1 -1
3
-2 -1 0
3
-2 -1 1
3
-2 -1 2
3
-2 0 -2
3
-2 0 -1
3
-2 0 0
3
-2 0 1
3
-2 0 2
3
-2 1 -2
3
-2 1 -1
3
-2 1 0
3
-2 1 1
3
-2 1 2
3
-2 2 -2
*/