前缀和 + 优化

P1114 “非常男女”计划

题目描述

近来，初一年的XXX小朋友致力于研究班上同学的配对问题（别想太多，仅是舞伴），通过各种推理和实验，他掌握了大量的实战经验。例如，据他观察，身高相近的人似乎比较合得来。

万圣节来临之际，XXX准备在学校策划一次大型的“非常男女”配对活动。对于这次活动的参与者，XXX有自己独特的选择方式。他希望能选择男女人数相等且身高都很接近的一些人。这种选择方式实现起来很简单。他让学校的所有人按照身高排成一排，然后从中选出连续的若干个人，使得这些人中男女人数相等。为了使活动更热闹，XXX当然希望他能选出的人越多越好。请编写程序告诉他，他最多可以选出多少人来。

输入格式

第一行有一个正整数nnn，代表学校的人数。n≤100000
第二行有nnn个用空格隔开的数，这些数只能是0或1，其中，0代表一个女生，1代表一个男生

输出格式

输出一个非负整数。这个数表示在输入数据中最长的一段男女人数相等的子序列长度。
如果不存在男女人数相等的子序列，请输出0。

输入输出样例
输入

9
0 1 0 0 0 1 1 0 0

输出

6

思路

前缀和+优化；
如果是女生，前缀和-1，否则前缀和+1；
优化：用了两重循环，即第二个循环从i+m开始，这样就可以保证之后找出来的符合条件的区间段一定比当前的最长区间段m长。

题解

#include<bits/stdc++.h>
using namespace std;
#define N 100005
int a[N],sum[N],n,m=0;

int main(){
	cin >> n;
	for(int i=1;i<=n;i++){
		cin >> a[i];
		if(a[i]==0)
			sum[i] = sum[i-1] - 1;
		else sum[i] = sum[i-1] + 1;
	}
	for(int i=1;i<n;i++){
		for(int j=i+m;j<=n;j++){
			if(sum[j]==sum[i-1])
				m = j - i + 1;
		}
	}
	cout << m << endl;	
	return 0;
}