每日一题2024年8月30日

E-不知道叫什么名字_2023年广东工业大学腾讯杯新生程序设计竞赛 (nowcoder.com)

大致思路：题目要求选择连续的，男女人数相等的。

题目给出 男生为1，女生为0，这里猜测要把女生的0改成-1（猜测，不知道对不对）
通过一些手段，前缀和+map，发现可以完美解决这个问题。

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef double db;
typedef long double ldb;
typedef pair<int, int> pii;
typedef pair<ll, ll> PII;
#define pb emplace_back
//#define int ll
#define all(a) a.begin(),a.end()
#define x first
#define y second
#define ps push_back
#define endl '\n'
#define IOS ios::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define lc u << 1
#define rc u << 1 | 1

void solve();

const int N = 1e6 + 10;
const int NN = 1e5;


signed main() {
    IOS;
    ll t = 1;
    while (t--)
        solve();
    return 0;
}

void solve() {
    ll n;
    cin >> n;
    vector<ll> a(n+1);
    for(int i = 1; i <= n; ++ i)
    {
        cin >> a[i];
        if(a[i] == 0)
        {
            a[i] = -1;
        }
    }
    vector<ll> sum(n+1);
    for(int i = 1; i <= n; ++ i)
    {
        sum[i] = a[i] + sum[i-1];
    }
    ll ans = 0;
    map<ll,ll> mp;
    mp[0] = 0;//最后一个是0的时候，这种情况没办法解决
    for(int i = 1; i <= n; ++ i)
    {
        if(mp.find(sum[i]) != mp.end())
        {
            continue;
        }
        mp[sum[i]] = i;
    }
    for(int i = 1; i <= n; ++ i)
    {
        ans = max(ans,i - mp[sum[i]]);
    }
    cout << ans << endl;
}