找相邻的子数组使其0和1的个数相等 Contiguous Array

最新推荐文章于 2024-08-27 16:49:08 发布

weixin_33997389

最新推荐文章于 2024-08-27 16:49:08 发布

阅读量217

点赞数

CC 4.0 BY-SA版权

原文链接：https://my.oschina.net/liyurong/blog/1604366

为什么80%的码农都做不了架构师？>>>

问题：

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.

解决：https://discuss.leetcode.com/topic/79906/easy-java-o-n-solution-presum-hashmap

① 找到最长的包含0，1（或1，0）的子串，让我们找相邻的子数组使其0和1的个数相等。对于求子数组的问题，我们需要时刻记着求累积和是一种很犀利的工具，但是这里怎么将子数组的和跟0和1的个数之间产生联系呢？我们需要用到一个trick，遇到1就加1，遇到0，就减1，这样如果某个子数组和为0，就说明0和1的个数相等。知道了这一点，我们用一个哈希表建立子数组之和跟结尾位置的坐标之间的映射。如果某个子数组之和在哈希表里存在了，说明当前子数组减去哈希表中存的那个子数字，得到的结果是中间一段子数组之和，必然为0，说明0和1的个数相等，我们更新结果res。。。

class Solution { //102ms
public int findMaxLength(int[] nums) {
for (int i = 0;i < nums.length;i ++){
if (nums[i] == 0) nums[i] = -1;
}
Map<Integer,Integer> map = new HashMap<>();
map.put(0,-1);
int sum = 0;
int max = 0;
for (int i = 0;i < nums.length;i ++){
sum += nums[i];
if (map.containsKey(sum)){
max = Math.max(max,i - map.get(sum));
}else {
map.put(sum,i);
}
}
return max;
}
}

② 在discuss中看到的。

class Solution { //45ms
public int findMaxLength(int[] nums) {
if (nums == null || nums.length == 0) return 0;
int[] map = new int[(2*nums.length) + 1];
Arrays.fill(map, -2);
map[nums.length] = -1;
int max = 0;
int sum = 0;
for (int i = 0;i < nums.length;i ++) {
sum = sum + (nums[i] == 0 ? -1 : 1);
if (map[sum + nums.length] >= -1) {
max = Math.max(max, i - map[sum + nums.length]);
} else {
map[sum + nums.length] = i;
}
}
return max;
}
}