洛谷 P1114 “非常男女”计划

洛谷 P1114 “非常男女”计划

Description

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

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

Input

  • 第一行有一个正整数n,代表学校的人数。n≤100000

    第二行有n个用空格隔开的数,这些数只能是0或1,其中,0代表一个女生,1代表一个男生

Output

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

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

Sample Input

9
0 1 0 0 0 1 1 0 0

Sample output

6

题解:

  • 我打的暴力。正解不是我这个(逃
  • 正常思路为O(n三方),加个前缀和优化变成O(n二方)。
  • 长度从大到小枚举,找到第一个符合的break
  • 奇数长度直接不用看continue掉
  • 加了这几个剪枝后就A了
#include 
#include 
#define maxn 100005
using namespace std;

int n;
int a[maxn], sum[maxn];

int read()
{
    int x = 0; char c = getchar();
    while(c < '0' || c > '9') c = getchar();
    while(c >= '0' && c <= '9') {x = x * 10 + c - '0'; c = getchar();}
    return x;
}

int main()
{
    cin >> n;
    for(int i = 1; i <= n; i++)
        a[i] = read(), sum[i] = sum[i - 1] + a[i];
    for(int len = n; len >= 0; len--)
        if(len % 2 == 0)
            for(int s = 1; s <= n - len + 1; s++)
                if((sum[s + len - 1] - sum[s - 1]) * 2 == len)
                {
                    cout << len;
                    return 0;
                }
    return 0;
}

转载于:https://www.cnblogs.com/BigYellowDog/p/11181922.html

你可能感兴趣的:(数据结构与算法)