【前缀和】| LeetCode 525题解 连续数组

一、题目链接

连续数组

二、题目

三、算法原理

四、编写代码

class Solution {
public:
    int findMaxLength(vector<int>& nums) 
    {
        unordered_map<int, int> hash;
        hash[0] = -1;// 默认有一个前缀和为 0 的情况
        
        int sum = 0, ret = 0;
        for (int i = 0; i < nums.size(); ++i)
        {
            sum += nums[i] == 0 ? -1 : 1;// 当前位置的前缀和
            if (hash.count(sum)) ret = max(ret, i - hash[sum]);
            else hash[sum] = i;// else是防止出现一样的<sum, i>存更大的下标
        }
        return ret;
    }
};