LeetCode 525 连续数组[前缀和 哈希表] HERODING的LeetCode之路

Posted 2021-06-17 HERODING23

解题思路：
又是一道前缀和+哈希表的每日一题，这里的前缀和需要取巧，即前缀和不是单纯的相加，而是遇到0加-1，遇到1加1，这样的好处是如果前缀和为0，说明从头到尾的0和1的个数相等，如果sum与之前的sum相等，说明两个sum之间的长度是符合1和0相等条件的串，这时候更新最长的长度，代码如下：

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        // 前缀和
        unordered_map<int, int> count;
        int sum = 0, maxLen = 0;
        // 初始位置标志为-1
        count[0] = -1;
        for(int i = 0; i < nums.size(); i ++) {
            // 0 加 -1,1加1
            sum += nums[i] ? -1 : 1;
            if(count.find(sum) != count.end()) {
                // 取最长
                maxLen = max(maxLen, i - count.find(sum) -> second);
            } else {
                // 更新sum的位置
                count[sum] = i;
            }
        }
        return maxLen;
    }
};


/*作者：heroding
链接：https://leetcode-cn.com/problems/contiguous-array/solution/cqian-zhui-he-ha-xi-biao-by-heroding-guea/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。*/