原题链接
这里我们引入相对差的概念。
很容易知道,如果第i个位置男女之差与第j个位置男女之差相等,那么第i+1到第j个位置男女人数相等。
所以,我们可以记录第i个位置的男女之差。最后遍历一遍就好了。
#include<iostream>
using namespace std;
int a[200005], l[200005], r[200005];
int max(int a, int b) {
return a > b ? a : b;
}
int main()
{
int n;
cin >> n;
int sum0 = 0, sum1 = 0;
for (int i = 1; i <= n; i++) {
int t;
cin >> t;
sum0 += (t == 0);
sum1 += (t == 1);
int x = sum0 - sum1 + n;
if (!l[x] && x != n)
l[x] = i;
else
r[x] = i;
}
int ans = 0;
for (int i = 0; i <= 2 * n; i++) {
ans = max(ans, r[i] - l[i]);
}
cout << ans;
return 0;
}