作者主页:苏州程序大白
作者简介:CSDN人工智能域优质创作者,苏州市凯捷智能科技有限公司创始之一,目前合作公司富士康、歌尔等几家新能源公司
如果文章对你有帮助,欢迎关注、点赞、收藏
有任何问题欢迎私信,看到会及时回复
关注苏州程序大白,分享粉丝福利
给定一个二进制数组 nums
, 找到含有相同数量的 0
和1
的最长连续子数组,并返回该子数组的长度。
示例 1:
输入: nums = [0,1]
输出: 2
说明: [0, 1] 是具有相同数量 0 和 1 的最长连续子数组。
示例 2:
输入: nums = [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量 0 和 1 的最长连续子数组。
提示:
1 <= nums.length <= 105
nums[i] 不是 0 就是 1
错误的思路:滑动窗口
,双指针
一开始打算运用双指针的想法,但是就实际而言是难以行通的。
对于[0,0,0,1,1,1,0]
最后可能为误判为0
。但是实际是6
总结一下,当不能明确得知可以淘汰的某些值时候就最好不要使用双指针。
前面那些0并不能直接判断是否需要淘汰,需要根据后面很多数来确定
public int findMaxLength(int[] nums) {
int res=0;
int n=nums.length;
int slow=0, fast=0;
int zeroNum=0, oneNum=0;
while (fast<n){
for (int i=0; i<2 && fast<n; i++){
if (nums[fast++]==0) zeroNum++;
else oneNum++;
}
if (oneNum==zeroNum) res = Math.max(res, fast-slow);
else {
if (nums[slow++]==0) zeroNum--;
else oneNum--;
}
}
return res;
}
前缀和+哈希表
把0当做-1处理,这里就转变为了求和为0的子数组的问题
key=前缀和,value=第一次出现过的位置
遍历数组,记录某个前缀和第一次出现的位置,设置初始值为map[0] = -1
当前缀和sum在此前出现过,则说明这两个前缀和区间差构成的所有元素和为0,满足条件更新res的值,否则记录到map中
class Solution {
public int findMaxLength(int[] nums) {
Map<Integer, Integer> map = new HashMap<>();
map.put(0, -1);
int sum = 0, res = 0;
for (int i = 0; i < nums.length; ++i) {
sum += nums[i] == 1 ? 1 : -1;
if (map.containsKey(sum)) {
res = Math.max(res, i - map.get(sum));
} else {
map.put(sum, i);
}
}
return res;
}
}