Leetcode.525 连续数组

最新推荐文章于 2024-07-25 11:15:54 发布

感觉画质不如…原神

最新推荐文章于 2024-07-25 11:15:54 发布

阅读量258

点赞数 1

分类专栏： Leetcode 文章标签： leetcode 算法哈希算法

本文链接：https://blog.csdn.net/m0_74396439/article/details/129585214

版权

Leetcode 专栏收录该内容

441 篇文章 2 订阅

订阅专栏

题目链接

Leetcode.525 连续数组 mid

题目描述

给定一个二进制数组 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

解法：前缀和 + 哈希表

把 1看作是 +1
把 0看作是 -1

将原问题转化为：和为 0的最长子数组的长度。

用一个哈希表 cnt记录sum出现的位置（cnt[0] = -1）。如果 sumi == sumj，那么说明 [i + 1 , j]区间和一定为 0，

我们只需要用 len来记录最大的长度即可。

时间复杂度： $O (n)$

C++代码：

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        int n = nums.size();
        unordered_map<int,int> cnt{{0,-1}};
        int len = 0;

        for(int j = 0 , s = 0;j < n;j++){
            s += (nums[j] == 1 ? 1 : -1);
            if(cnt.count(s)){
                int i = cnt[s];
                if(j - i > len) len = j - i;
            }
            else cnt[s] = j;
        }
        return len;
    }
};

Python代码：

class Solution:
    def findMaxLength(self, nums: List[int]) -> int:
        n = len(nums)
        cnt , s = {0:-1} , 0

        ans = 0

        for j in range(0,n):
            s = s + 1 if nums[j] == 1 else s - 1
            if s in cnt:
                ans = max(ans , j - cnt[s])
            else:
                cnt[s] = j
        return ans