艰难的选择

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

Com_man_der

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

阅读量316

点赞数

CC 4.0 BY-SA版权

分类专栏：递推

本文链接：https://blog.youkuaiyun.com/Com_man_der/article/details/86628426

递推专栏收录该内容

13 篇文章

订阅专栏

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。

题解：把男生（0）变成-1，女生仍是1

f[i]为a数组的前缀和

若f[i] - f[j-1]=0 那么i~j区间是合法的

f[i] - f[j-1]=0

f[i] = f[j-1]

问题就转化成为，在1~i-1找一个数和当前数一样。

一个桶就行了。

#include<bits/stdc++.h>
#define N 100010
#define inf 2147483647
#define ll long long
#define point(a) multiset<a>::iterator 
#define mod (ll)(500000)
#define mem(a,b) memset(a,b,sizeof (a))
#define open(x) freopen(x".in","r",stdin);freopen(x".out","w",stdout);
using namespace std;

const int fix = N-10;
int n,i,j,t,ans;
int a[N],f[N];
int bz[N*2];

int main()
{
	open("choose");
	scanf("%d",&n);
	for(i=1;i<=n;i++)scanf("%d",&a[i]),a[i]=a[i]==1 ? 1:-1;
	for(i=1;i<=n;i++)f[i] = f[i-1]+a[i];
	mem(bz,-1);
	bz[0+fix]=0;
	bz[f[1]+fix]=1;
	for(i=2;i<=n;i++)
	{
		if(bz[f[i]+fix]!=-1) ans=max(ans ,  i - bz[f[i]+fix]); else
		 	bz[f[i]+fix]=i;
	}
	printf("%d\n",ans);
	return 0;
}

时间复杂度：

O（n）