525. Contiguous Array

来源:互联网 发布:gson解析json数组 编辑:程序博客网 时间:2024/06/05 17:18

给定一个数组,只包含0和1,求最大的子数组长度,使得0和1的数目相等。

思路:

这个问题可以转变成560 最长区间和为给定值这道题。只需要将0改成-1.建立一个哈希标保存,累加和是一个值的最左位置。当前位置i的累加和是x,那么只要找到最左边累加和也是x的位置j,那么(j,i]的累加和就是0,说明0和1的数目相等。

这个题注意两点:

1、加和为0的位置设置成-1;

2、由于map的默认值是0,所以第一个位置的值的加和对应的位置是0,这和判断有没有这个键值是混淆的,需要单独判断当前和是否是第一个位置的值

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


原创粉丝点击