这段代码实现了用前缀和和哈希表寻找0和1数量相同的最长子数组。核心思路是将0视为-1,然后计算前缀和,利用哈希表记录每个前缀和第一次出现的位置,当遇到相同前缀和时计算长度并更新最大值。
关键步骤:
- 初始化哈希表,记录前缀和0的初始位置为-1
- 遍历数组,遇到1时前缀和加1,遇到0时前缀和减1
- 检查当前前缀和是否已存在哈希表中:
- 若存在,计算当前位置与首次出现位置的距离,更新最大长度
- 若不存在,记录当前前缀和的首次出现位置
- 返回最大长度
示例输入[0,1,0,1]的输出为4,对应整个数组。时间复杂度O(n),空间复杂度O(n)。
cpp
#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;
}