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的位置为-1hashmap[0] = -1;for (int i = 0; i < nums.size(); i++) {// 遇到1加1，遇到0减1if (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;
}