jzoj3913. 艰难的选择（B组——Day6）

最新推荐文章于 2020-07-26 21:50:16 发布

原创最新推荐文章于 2020-07-26 21:50:16 发布 · 291 阅读

0 ·

CC 4.0 BY-SA版权

jzoj 同时被 3 个专栏收录

94 篇文章

订阅专栏

暴力模拟

34 篇文章

订阅专栏

前缀和

3 篇文章

订阅专栏

本文介绍了一道算法竞赛题目“艰难的选择”的解决方案，该题要求从一系列按身高排列的人中选出男女人数相等且人数最多的连续子序列。通过使用前缀和的概念和优化技巧，提供了一个高效的算法实现。

jzoj3913. 艰难的选择（B组——Day6）

题目

Description

Yzx已经当过多次“媒人”了。他因此获得了许多经验。例如，距Yzx观察，身高相近的人似乎比较合得来。
Yzx在学校策划了一次大型的“非常男女”配对活动。对于这次活动的参与者，Yzx有自己独特的选择方式。他希望能选择男女人数相等且身高都很接近的一些人。这种选择方式实现起来很简单，他让学校的所有人按照身高排成一排，然后从中选出连续的若干个人，使得这些人中男女人数相等。Yzx当然希望他能选出的人越多越好，请告诉他最多可以选出多少人来。

Input

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

Output

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

Sample Input

9
0 1 0 0 0 1 1 0 0

Sample Output

Data Constraint

30%的数据，n<=100。
50%的数据，n<=1000。
100%的数据，n<=100000。

解析

前缀和+优化

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;

int n,maxn,k;
int a[100010];

int main(){
	scanf("%d",&n);
	for (int i=1;i<=n;i++){
		scanf("%d",&k);
		if (k==0) a[i]=a[i-1]-1;
			else a[i]=a[i-1]+1;
	}
	k=0;
	while (maxn<=n-k){
		int i=k+maxn+2;
		while (i<=n){
			if (a[i]-a[k]==0) maxn=max(maxn,i-k);
			i+=2;
		}
		k++;
	}
	printf("%d",maxn);
	return 0;
}