1.525. 连续数组 - 力扣(LeetCode)
在我对通义千问的一番折磨下,终于弄清楚一点点了。哈希表存储前缀和数组值
用一个counter来记录nums中0、1数量差值的变化。 哈希表map存储某个特定的counter值首次出现的位置。counter的计算:出现1就加1,出现0就减1。第二次出现同样的counter值意味着从第一次出现的counter位置到第二次出现的counter位置中间出现的0和1的次数一样了 才会导致同样的counter值出现第二次。平衡了平衡了。为什么map[0] = - 1 而不是map[0] = 0呢,我理解的还不是很清白。是为了第一个位置开始就平衡的情况。
class Solution {
public:
int findMaxLength(vector<int>& nums) {
int counter = 0, res = 0;
unordered_map<int, int> map;
map[0] = - 1;
for(int i = 0; i < nums.size(); i++){
if(nums[i] == 1)
counter++;
else
counter--;
if(map.count(counter))
res = max(res, i - map[counter]);
else
map[counter] = i;
}
return res;
}
};