连续的01抵消最长段

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

输入格式：
第一行有一个正整数 n，代表学校的人数。
第二行有 n 个用空格隔开的数，这些数只能是 0 或 1，其中，0 代表一个女生，1代表一个男生。

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

样例输入：
9
0 1 0 0 0 1 1 0 0

样例输出：
6

说明：前缀和

#include <algorithm>
#include <iostream>
using namespace std;
int vis[200005];//索引代表前缀和，值（01）代表是否有对应的前缀和 
int main() {
    int n;
    cin >> n;
    int ans = 0, sum = 100000;//初始前缀和设为100000，不能设为0，不然数组下标可能会出现负数 
    for (int i = 1; i <= n; i++) {//i代表同学编号 
        int x;
        cin >> x;
        if (x == 1) {//求对应同学的前缀和 
            sum++;
        } else {
            sum--;
        }
        if (sum == 100000) {
            ans = i;
        } else if (vis[sum] == 0) {//如果还没有编号对应此sum前缀和，就将前缀和与编号对应起来 （sum:前缀和。i:编号） 
            vis[sum] = i; 
        } else {//如果已经有对应的前缀和了，就将最早的那个编号的人编号求差算出间隔，并与以往最大间隔比较求最大间隔 
            ans = max(ans, i - vis[sum]);
        }
    }
    cout << ans << endl;
    return 0;
}