[前缀和] Jzoj P3913 艰难的选择

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

6

 

Data Constraint

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

题解

• 这题挺水的，跑一个前缀和
• 那么就可以暴力枚举长度，然后暴力判断（长度与前缀和的差是两倍关系）
• 最后，加个快读或吸一口臭氧就可以过了

代码

#include <cstdio>
#include <iostream>
#define N 100010
using namespace std;
int n,ans,a[N],sum[N];
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++) scanf("%d",&a[i]),sum[i]=sum[i-1]+a[i];
    for (int i=1;i<=n-ans;i++)
        for (int j=i+ans+1;j<=n;j+=2) 
            if ((sum[j]-sum[i-1])*2==j-i+1)    ans=max(ans,j-i+1);
    printf("%d",ans);
}