cpp
#include<bits/stdc++.h>
using namespace std;
const int N=2e5+10;
int a[N];
long long s[N];
void solve()
{
int n;
cin>>n;
vector<int> zero;
for(int i=1;i<=n;i++)
{
cin>>a[i];
s[i]=s[i-1]+a[i];
if(a[i]==0)
zero.push_back(i);
}
zero.push_back(n+1);
long long ans=0;
for(int i=1;i<zero[0];i++)
if(s[i]==0)
ans++;
for(int i=0;i+1<zero.size();i++)
{
int mx=0;
map<long long,int> mp;
for(int j=zero[i];j<zero[i+1];j++)
mx=max(mx,++mp[s[j]]);
ans+=mx;
}
cout<<ans<<endl;
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(nullptr);
int t;
cin>>t;
while(t--)
solve();
return 0;
}
贪心
分板块处理数组,每一个板块贪心求最大值,有一些前缀和,差分的应用,需要一定的代码实现能力, m a p map map 计数
题目阅读理解比较困难,题意是,数组有一个前缀和是 0 0 0 就表示 1 1 1 分,可以对每一个 0 0 0 进行操作,求分数的最大值
差分有一个性质,改变差分数组的一个数字,会影响该数字后面的所有数字(包括该数字)(影响的是差分数组的前缀和数组)
输入的数组可以看成是前缀和数组的差分数组
划分的界限是 0 0 0 ,因为可以进行操作的只有 0 0 0 这个元素,在某个 0 0 0 处加上一个数字,在下一个 0 0 0 处减去该数字,就可以保证该操作只对两个 0 0 0 中间的前缀和产生了影响
单独考虑第一个 0 0 0 之前的所有元素
然后分块考虑中间的两个零之间的元素
最后把所有的贡献求和
因为涉及到 2 ∗ 1 0 5 2*10^5 2∗105 个数字的相加,所以会超过 i n t int int 的数据范围,前缀和数组需要使用 l o n g l o n g long\ long long long ,答案也需要使用 l o n g l o n g long\ long long long ,每一次贪心的是,出现次数最多的前缀和元素,修改零对应位置的差分数组数值,可以使得出现次数最多的前缀和元素变成零,该情况可以使得答案取到最大值
涉及到边界问题,所以保存零的位置的时候多存了一个零的位置,相当于用一个 v e c t o r vector vector 把分界点存了下来,时间复杂度还是没有改变,还是只需要把原数组遍历一遍,分开贪心求贡献,贡献求和是答案