P1114 “非常男女”计划

题目描述

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

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

输入格式

第一行有一个正整数 nn，代表学校的人数。n \le 100000n≤100000

第二行有 nn 个用空格隔开的数，这些数只能是 00 或 11，其中，00 代表一个女生，11 代表一个男生

输出格式

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

如果不存在男女人数相等的子序列，请输出 00。

输入输出样例

输入 #1复制

9
0 1 0 0 0 1 1 0 0

输出 #1复制

6

#include<bits/stdc++.h>
#include<algorithm>
#define MAX 100005
#include<stdio.h>
using namespace std;
struct place{
	int num,pos;
}arr[MAX];
bool cmp(place x,place y){
	if(x.num!=y.num) return x.num<y.num;
	return x.pos>y.pos;//相同前缀和的被相邻排放 
}
int Max(int x,int y){
	return x>y? x:y;
}
int main(){
	int n,temp,i;
	scanf("%d",&n);
	for(i=1;i<=n;i++){
		scanf("%d",&temp);
		if(temp==1) arr[i].num=1;//如果是男生，标记为1 
		else arr[i].num=-1;//女生，标记为0 
		if(n!=1) arr[i].num+=arr[i-1].num;//前缀和 
		arr[i].pos=i;//位置 
	}
	sort(arr+1,arr+1+n,cmp);
	int j,res=0;
	for(i=1;i<=n;i++){
		for(j=i+1;j<=n&&arr[i].num==arr[j].num;j++){
			res=Max(res,arr[i].pos-arr[j].pos);
			//当l位置的前缀和的值和r位置的前缀和的值相等时
			//说明此区间(l,r]内的男女人数一样
			//这个区间一共有r-l，不要搞成r-l+1 
			if(arr[i].num==0) res=Max(res,arr[i].pos);
			//特殊情况，当前缀和为0，意味着从开始到当前位置男女一样，当前位置就是最大值  
	}
		i=j-1;//num为下一数值 
	}
	printf("%d",res);
}

tle的答案 

#include <cstdio>
using namespace std;
const int MAXN = 1e5+4;
int nums[MAXN];//原始数据
int qzh[MAXN];//前缀和
int main() {
    int n;
    scanf("%d", &n);
 	for (int i=1; i<=n; i++) {
        scanf("%d", &nums[i]);//读取0/1 
        qzh[i] = qzh[i-1]+nums[i];
    }
    int m= (1==n%2) ? n-1 : n;
    for (int i=m; i>=2; i-=2) {//i代表区间长度 
        //n,n-2,n-4,...,4,2 的长度进行验证
        for (int j=n; j-i>=0; j--) {//j代表区间起点 
            if ((qzh[j]-qzh[j-i])*2==i) {
            	//如果此区间内男女人数相等，前缀和之差乘2应该等于这段区间的长度 
                printf("%d\n", i);
                return 0;
            }
        }
    }
    printf("0\n");
 
    return 0;
}