洛谷——P1114 “非常男女”计划

原题链接
这里我们引入相对差的概念。
很容易知道,如果第i个位置男女之差与第j个位置男女之差相等,那么第i+1到第j个位置男女人数相等。
所以,我们可以记录第i个位置的男女之差。最后遍历一遍就好了。

#include
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;
}

你可能感兴趣的:(洛谷,C++)