Given a binary array, find the maximum length of a contiguous subarray with equal number of 0 and 1.
Example 1:
Input: [0,1] Output: 2 Explanation: [0, 1] is the longest contiguous subarray with equal number of 0 and 1.
Example 2:
Input: [0,1,0] Output: 2 Explanation: [0, 1] (or [1, 0]) is a longest contiguous subarray with equal number of 0 and 1.
Note: The length of the given binary array will not exceed 50,000.
思路:给定一个二进制数组,找出0和1的个数相等的长度最大子数组,并返回这个子数组的长度。换一种思维,在这个字串中要求0和1个数相等,若是将数组中的所有0用-1来替换,那么这个子数组就满足所有元素相加和为0,问题就转换成了我们要找一个长度最大的子数组,这个子数组中的所有数字相加和为0,并返回这个字数组的长度。使用sum来记录循环中每次累加一个元素后的结果,使用map来记录sum的值(第一维)以及对应的数组下标(第二维),循环中每次累加后都会在map中寻找是否出现过相等的sum,如果相等,那么就对最大长度子数组进行更新,若是没有出现过这个sum,则更新map,将其存入map中。举个例子:比如说之前有过一个sum = 2,现在又出现sum = 2了,说明在第一次sum等于2的时候,它前面所有元素加起来总和是2,那么在它前面去掉2个元素1就能满足0,同理当前的sum = 2也可以通过去掉最前面的2个元素1使sum = 0,所以看看j - ma[sum]是否比之前的最大值大,如果比之前最大值大就更新最大值。
class Solution {
public:
int findMaxLength(vector<int>& nums) {
map<int,int> ma;
int result=0,sum=0;
ma[0]=-1;
for(int i=0;i!=nums.size();++i)
if(nums[i]==0)
nums[i]=-1;
for(int j=0;j!=nums.size();++j)
{
sum+=nums[j];
if(ma.find(sum)!=ma.end())
result=max(result,j-ma[sum]);
else
ma[sum]=j;
}
return result;
}
};
reference:http://blog.csdn.net/liuchuo/article/details/56282576