C语言编程题：“非常男女”计划（C语言版）

1.题目描述

展开近来，初一年的 xxx 小朋友致力于研究班上同学的配对问题(别想太多，仅是舞伴)，通过各种推理和实!验，他掌握了大量的实战经验。例如，据他观察，身高相近的人似乎比较合得来。
万圣节来临之际， xxx 准备在学校策划一次大型的“非常男女”配对活动。对于这次活动的参与者， xxx有自己独特的选择方式。他希望能选择男女人数相等且身高都很接近的一些人。这种选择方式实现起来很简单。他让学校的所有人按照身高排成一排，然后从中选出连续的若干个人，使得这些人中男女人数相等。为了使活动更热闹， xx当然希望他能选出的人越多越好。请编写程序告诉他，他最多可以选出多少人来。

2.输入格式

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

3.输出格式

输出一个非负整数。这个数表示在输入数据中最长的一段男女人数相等的子区间的长度,如果不存在男女人数相等的子区间，请输出 0。

输入输出样例

输入：

9
0 1 0 0 0 1 1 0 0

输出：

6

代码：

#include <stdio.h>
#include <stdlib.h>

int main() {
    int n;
    scanf("%d", &n);
    
    int *arr = (int *)malloc(n * sizeof(int));
    for (int i = 0; i < n; i++) {
        scanf("%d", &arr[i]);
        // 转换女生（0）为 -1
        if (arr[i] == 0) {
            arr[i] = -1;
        }
    }

    // 用于记录每个前缀和首次出现的索引
    int *prefixIndex = (int *)malloc((2 * n + 1) * sizeof(int));
    for (int i = 0; i < 2 * n + 1; i++) {
        prefixIndex[i] = -2; // 初始化为 -2 表示未出现
    }
    prefixIndex[n] = -1; // 前缀和为 0 在索引 -1 处出现

    int maxLength = 0;
    int prefixSum = 0;

    for (int i = 0; i < n; i++) {
        prefixSum += arr[i];
        // adjust the prefixSum to handle negative indices
        if (prefixIndex[prefixSum + n] == -2) { 
            prefixIndex[prefixSum + n] = i; // 存储第一次出现的索引
        } else {
            int length = i - prefixIndex[prefixSum + n]; // 计算长度
            if (length > maxLength) {
                maxLength = length; // 更新最大长度
            }
        }
    }

    printf("%d\n", maxLength);
    
    free(arr);
    free(prefixIndex);
    return 0;
}