LeetCode 525 连续数组

题目链接:LeetCode 525 连续数组

题目大意:
给定一个二进制数组nums,找到含有相同数量的01的最长连续子数组,并返回该子数组的长度。

题解:
0换作1,则问题变为求最长的连续子数组,其元素和为0
首先计算数组的前缀和,设preSum[j]preSum[i]=0,则表示存在长为ji的连续子数组,其元素和为0
可以用哈希表存储第一次出现的前缀和,之后再次出现相等的前缀和即可直接计算子数组长度。

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        int sum = 0, ans = 0;
        unordered_map<int, int> hashTable;
        hashTable[0] = -1;
        for (int i = 0; i < nums.size(); ++i) {
            sum += (nums[i] ? 1 : -1);
            if (hashTable.count(sum)) {
                ans = max(ans, i - hashTable[sum]);
            } else {
                hashTable[sum] = i;
            }
        }
        return ans;
    }
};
posted @   ZZHHOOUU  阅读(19)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 终于写完轮子一部分:tcp代理 了,记录一下
· 震惊!C++程序真的从main开始吗?99%的程序员都答错了
· 别再用vector<bool>了!Google高级工程师:这可能是STL最大的设计失误
· 单元测试从入门到精通
· 【硬核科普】Trae如何「偷看」你的代码?零基础破解AI编程运行原理
点击右上角即可分享
微信分享提示