给定一个数组,只包含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;
}
};