Leetcode 525. 连续数组（将0变为-1转化为求和为0的最长区间子数组）

最新推荐文章于 2021-11-20 12:57:04 发布

wwxy261

最新推荐文章于 2021-11-20 12:57:04 发布

阅读量431

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/wwxy1995/article/details/117495169

版权

算法专栏收录该内容

3633 篇文章 122 订阅

订阅专栏

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