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.
题目:给定一个01串,只含有0和1,求最长的的连续子串,并且子串的0和1的数目相等。
在解决这个题之前,我们先考虑另外一个问题:给定一个序列,其中有正数、负数和0。请你求出和为aim的最长连续子串。
对于这个问题,我们分析一下,假如从序列的第一个数累加到第i个数的和为K,那么我们先在1-i中找到累加和为(K-aim)的最先出现的位置,从这个位置到i的累加和就是aim。也就是说,我们可以找到以第i个位置结尾的累加和为aim的最长子串,把i依次遍历一遍,我们就能取到最长的连续子串了。
01相等子串这个问题也可以用上述的思想来解决,需要做一点小小的改动,就是把01串中的0改成-1,这样一来,题目就变成了求和为0的最长子串。
代码实现:
class Solution {
public int findMaxLength(int[] nums) {
HashMap<Integer,Integer> hm = new HashMap<>();
int max = 0;
int sum = 0;
hm.put(0,-1);
for(int i=0;i<nums.length;i++){
if(nums[i] == 0)
nums[i]--;
sum += nums[i];
if(hm.containsKey(sum)){
max = Math.max(max, i-hm.get(sum));
}else
hm.put(sum, i);
}
return max;
}
}