#include <iostream>
#include <unordered_map>
#include <vector>
using namespace std;

int main() {
    int n;
    cin >> n;
    vector<int> a(n);
    for (int i = 0; i < n; ++i) {
        cin >> a[i];
        a[i] = a[i] == 0 ? -1 : 1;  // 将0和1转换为-1和1
    }
    
    unordered_map<int, int> sum_pos;
    int sum = 0;
    int max_len = 0;
    
    for (int i = 0; i < n; ++i) {
        sum += a[i];
        if (sum == 0) {
            max_len = i + 1;  // 从开头到当前位置满足条件
        }
        if (sum_pos.find(sum) != sum_pos.end()) {
            max_len = max(max_len, i - sum_pos[sum]);
        } else {
            sum_pos[sum] = i;
        }
    }
    
    cout << max_len << endl;
    return 0;
}

#include #include <unordered_map> #include

using namespace std;

int find_max_equal_subsequence_length(int n, const vector& arr) { // 将0转换为-1 vector modifiedArr(arr.begin(), arr.end()); for (int i = 0; i < n; ++i) { if (modifiedArr[i] == 0) { modifiedArr[i] = -1; } }

// 计算前缀和数组
vector<int> prefixSum(n + 1, 0);
for (int i = 1; i <= n; ++i) {
    prefixSum[i] = prefixSum[i - 1] + modifiedArr[i - 1];
}

// 创建一个unordered_map，用于存储每个前缀和第一次出现的位置
unordered_map<int, int> prefixSumIndex;

int maxLength = 0;

for (int i = 0; i <= n; ++i) {
    // 如果当前前缀和已经在unordered_map中出现过，说明两次出现的位置之间的子数组男女人数相等
    if (prefixSumIndex.find(prefixSum[i]) != prefixSumIndex.end()) {
        maxLength = max(maxLength, i - prefixSumIndex[prefixSum[i]]);
    } else {
        // 如果当前前缀和第一次出现，记录其位置
        prefixSumIndex[prefixSum[i]] = i;
    }
}

return maxLength;

}

int main() { // 读取输入 int n; cin >> n;

vector<int> arr(n);
for (int i = 0; i < n; ++i) {
    cin >> arr[i];
}

// 计算并输出结果
int result = find_max_equal_subsequence_length(n, arr);
cout << result << endl;

return 0;

}

• s[i] 表示前缀和
• s[i] - s[j] 表示 j + 1 到 i 之间 1 的个数
• 那么 $2*(s[i] - s[j]) = i - j$
• $2*s[i]-i=2*s[j]-j$
• 我们用 $b[i]$ 表示 $2*s[i] - i$