Zero-Sum Prefixes

题意

给你一个长度为n的数组，你可以把值为0的数改变成任意一个数字，操作次数不限制，问有多少个前缀和为0。

思路

首先我们要清楚，每个0所处的位置，对他前面的前缀和是没有任何影响的，只会对其后的前缀和产生影响，我们只需要统计每两个0之间，哪个前缀和出现的最多，答案累加就行了，对于第一个0之前的每个前缀和再单独判断就行了。

代码

#include<bits/stdc++.h>

#define IOS ios::sync_with_stdio(false);cin.tie(nullptr)
#define int long long 
#define endl "\n"

using namespace std;

const int N = 2e5 + 10;

int n, m, _;

void solve()
{
    cin >> n;
    vector<int> sum(n+1), a(n), zero;
    
    for(int i = 0; i < n; i ++)
    {
        cin >> a[i];
        sum[i+1] = sum[i]+a[i];
        if(!a[i]) zero.push_back(i+1);
    }
    
    zero.push_back(n+1);
    
    map<int, int> mp;
    int ans = 0;
    for(int i = 1; i < zero.size(); i ++)
    {
        mp.clear();
        int ma = 0;
        for(int j = zero[i-1]; j < zero[i]; j ++)//判断每两个0之间出现次数最多的前缀和
        {
            ma = max(ma, ++mp[sum[j]]);
        }
        ans += ma;
    }
    
    for(int i = 1; i < zero[0]; i ++)//单独判断第一个0之前的前缀和出现的0次数
    {
        if(!sum[i]) ++ans;
    }
    
    cout << ans << endl;
}

signed main()
{
    IOS;
    cin >> _;
    while(_--)
        solve();
    return 0;
}