连续数组
难度:
标签:
题目描述
English description is not available for the problem. Please switch to Chinese.
代码结果
运行时间: 107 ms, 内存: 23.7 MB
/*
思路:
Java Stream 不能直接处理这个问题,因为我们需要跟踪累积和的下标。因此,我们在这里不使用 Stream,而是尽量简洁地使用常规方法。
步骤与上面相同。
*/
import java.util.HashMap;
import java.util.stream.IntStream;
public class Solution {
public int findMaxLength(int[] nums) {
HashMap<Integer, Integer> map = new HashMap<>();
map.put(0, -1); // 初始化累积和为0的位置为-1
int[] count = {0};
int[] maxLength = {0};
IntStream.range(0, nums.length).forEach(i -> {
count[0] += (nums[i] == 1) ? 1 : -1;
if (map.containsKey(count[0])) {
maxLength[0] = Math.max(maxLength[0], i - map.get(count[0]));
} else {
map.put(count[0], i);
}
});
return maxLength[0];
}
}
解释
方法:
该题解采用了哈希表记录前缀和的方法。定义cur为当前前缀和,遍历数组nums时,遇到1则cur加1,遇到0则cur减1。用一个字典dic记录cur第一次出现的位置和最后一次出现的位置。最后,遍历dic中的所有值,计算最大的位置差值,即为含有相同数量的0和1的最长连续子数组的长度。
时间复杂度:
O(n)
空间复杂度:
O(n)
代码细节讲解
🦆
为什么在遍历数组时选择用前缀和的方式表示0和1的数量关系?
▷🦆
字典中存储前缀和的第一次和最后一次出现的位置,这种方法能够确保计算的子数组确实是0和1数量相等吗?
▷🦆
代码中对于前缀和为0的特殊处理`res = max(res, dic[0][1] + 1)`的逻辑是什么?为什么要加1?
▷