leetcode 525. 连续数组

这段代码实现了用前缀和和哈希表寻找0和1数量相同的最长子数组。核心思路是将0视为-1，然后计算前缀和，利用哈希表记录每个前缀和第一次出现的位置，当遇到相同前缀和时计算长度并更新最大值。

关键步骤：

1. 初始化哈希表，记录前缀和0的初始位置为-1
2. 遍历数组，遇到1时前缀和加1，遇到0时前缀和减1
3. 检查当前前缀和是否已存在哈希表中：
   • 若存在，计算当前位置与首次出现位置的距离，更新最大长度
   • 若不存在，记录当前前缀和的首次出现位置
4. 返回最大长度

示例输入0,1,0,1的输出为4，对应整个数组。时间复杂度O(n)，空间复杂度O(n)。

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

int findMaxLength(vector<int>& nums) {
    int maxLen = 0;
    int counter = 0;
    // 哈希表记录前缀和第一次出现的位置
    unordered_map<int, int> hashmap;
    // 初始化前缀和为0的位置为-1
    hashmap[0] = -1;

    for (int i = 0; i < nums.size(); i++) {
        // 遇到1加1，遇到0减1
        if (nums[i] == 1) {
            counter++;
        } else {
            counter--;
        }

        // 如果当前前缀和已经存在，计算当前位置到首次出现位置的长度
        if (hashmap.find(counter) != hashmap.end()) {
            int len = i - hashmap[counter];
            if (len > maxLen) {
                maxLen = len;
            }
        } else {
            // 记录前缀和首次出现的位置
            hashmap[counter] = i;
        }
    }

    return maxLen;
}

int main() {
    vector<int> nums = {0, 1, 0, 1};
    cout << "最长子数组长度: " << findMaxLength(nums) << endl;
    return 0;
}