365天挑战LeetCode1000题——Day 261 前缀和 + 哈希表

ShowM3TheCode

于 2023-03-09 15:00:54 发布

阅读量134

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode千题之路文章标签：散列表 leetcode 算法

本文链接：https://blog.youkuaiyun.com/ShowMeTheCod3/article/details/129423115

LeetCode千题之路专栏收录该内容

124 篇文章

订阅专栏

该代码片段展示了一个C++解决方案，用于计算给定0和1数组中具有相同数量的1和0的最长连续子数组的长度。它使用哈希映射存储当前计数值及其前一次出现的位置，从而有效地找到最大长度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

记录每个位置的1和0的差的个数，如果以前出现过这个数，那么答案有可能就是二者的差。

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        int maxLength = 0;
        unordered_map<int, int> mp;
        int cnt = 0;
        mp[cnt] = -1;
        int i = 0;
        for (int x : nums) {
            cnt += x == 1 ? 1 : -1;
            if (mp.count(cnt)) maxLength = max(maxLength, i - mp[cnt]);
            else mp[cnt] = i;
            i++;
        }
        return maxLength;
    }
};