每日OJ题_算法_前缀和⑦_力扣525. 连续数组

目录

力扣525. 连续数组

解析代码

力扣525. 连续数组

525. 连续数组

难度 中等

给定一个二进制数组 nums , 找到含有相同数量的 0 和 1 的最长连续子数组，并返回该子数组的长度。

示例 1:

输入: nums = [0,1]
输出: 2
说明: [0, 1] 是具有相同数量 0 和 1 的最长连续子数组。

示例 2:

输入: nums = [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。

提示：

• 1 <= nums.length <= 10^5
• nums[i] 不是 0 就是 1

class Solution {
public:
    int findMaxLength(vector<int>& nums) {

    }
};

解析代码

把数组的0变成/当成-1之后，就转化成力扣560. 和为 K 的子数组了（此时K=0）。

​
class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        // 转化成找一段和为0的最长子数组 -> 把数组的0变成/当成-1
        int sum = 0, n = nums.size(), ret = 0;
        unordered_map<int, int> hash(n); // 左存前缀和，右存下标
        hash[0] = -1; // 默认有一个前缀和为0的情况，下标为-1，如[0,1]返回就是1 - (-1) = 2
        for(int i = 0; i < n; ++i)
        {
            sum += nums[i] == 0 ? -1 : 1;
            if(hash.count(sum)) // 找最左面有没有前缀和和此时一样的
            {
                ret = max(ret, i - hash[sum]); // 找到前面出现一样的，则那个位置到现在的位置和为0
            }
            else // 前面没有一样的才更新，保证最左边
            {
                hash[sum] = i; 
            }
        }
        return ret;
    }
};