525. 连续数组

前缀和周不用说了！还以为是dp嘞，看来连续子数组不仅要考虑dp还要考虑prefixsum
我们把0看成-1，那么就变成了求一段和为0的最长解，所以我们用前缀和+哈希来做。前缀和是防止重复求和，哈希是为了快速查找相同是否有相同前缀和的情况，如果有，中间差的那一段一定是0！
每一轮循环先更新前两位的，不能更新前一位的哦，不然要是和sum[i]相同，那么这一段连续子数组和可能是只有1个的，虽然不可能发生，但是还是要注意。
直接判断前两位是否已存在，如果不存在就更新一下前两位的前缀和，因为要保证连续子数组长度>=2，所以我们在上面先更新前两位的，然后再来检测这一位的

class Solution {
public:
    int findMaxLength(vector<int>& nums) {
        //可能新来一个会打破，但是不能保证加上后面的会让它重回平衡
        //先暴力试试
        // int res = 0;
        // for(int i = 0; i < nums.size()-1; ++i){
        //     int tmp = nums[i] == 0?-1:1;
        //     for(int j = i + 1; j < nums.size(); ++j){
        //         tmp += nums[j] == 0?-1:1;
        //         if(tmp == 0) res = max(res,j-i+1);
        //     }
        // }
        // return res;

        //我们把0看成-1，那么就变成了求一段和为0的最长解
        //和昨天差不多
        int n = nums.size();
        int res = 0;
        vector<int> sum(n + 1);
        //sum[i]是nums[0]到nums[i-1]的和
        for(int i = 1; i <= n; ++i){
            sum[i] = sum[i-1] + (nums[i-1] == 0 ?-1:1);
        }
        //key记录前缀和大小，value记录是从0到第几位
        unordered_map<int,int> ma;
        //然后将每一位的前缀和加入map中，到后面两位以上的前缀和的时候判断是否有这一位前缀和出现过
        for(int i = 2; i <= n; ++i){
            //直接判断前两位是否已存在，如果不存在就更新一下前两位的前缀和，因为要保证连续子数组长度>=2，所以我们在上面先更新前两位的，然后再用前两位的来检测这一位的
            if(!ma.count(sum[i-2])) ma[sum[i-2]] = i-2;
            if(ma.count(sum[i])) res = max(res,i-ma[sum[i]]);
        }
        return res;
    }
};