洛谷-P1114 “非常男女”计划（java）

题目描述

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

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

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

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

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

如果不存在男女人数相等的子序列，请输出0。
输入输出样例
输入 #1
9
0 1 0 0 0 1 1 0 0
输出 #1
6

思路：

女生为-1，男生为1，人数相等时，两个值相减等于0，求两个数相减为0的最大区间。

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        int n = input.nextInt();
        int[] student = new int[n+1];
         //因为数组第一个我们不用,所以数组长度为n+1
         
        int index = 0,flag;
        // index为这个最大区间的值,flag为暂存输入的数
        
        for (int i = 1; i <= n; i++) {
            flag = input.nextInt();
            if (flag == 1){
                student[i] = student[i-1] + 1;//是男则加一
            }else {
                student[i] = student[i-1] - 1;//是女则减一

            }

        }
        for (int i = 1;i <= n - index; i ++){ 
            for (int j = i + index; j <= n;j++){
                if (student[j]-student[i-1]==0){
                    index = j-i+1;
                    //重点:!当两个数相等时,即男生与女生相等,下次遍历的时候,区间幅度应大于当前区间幅度index.
                }
            }
        }
        System.out.println(index);
    }
}

题目原链接:https://www.luogu.com.cn/problem/P1114