#include <iostream> #include <unordered_map> #include <vector>

using namespace std;

int find_max_equal_subsequence_length(int n, const vector<int>& arr) { // 将0转换为-1 vector<int> 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$