LeetCode刷题实战525:连续数组
共 1397字,需浏览 3分钟
·
2022-02-16 23:28
Given a binary array nums, return the maximum length of a contiguous subarray with an equal number of 0 and 1.
示例
示例 1:
输入: nums = [0,1]
输出: 2
说明: [0, 1] 是具有相同数量 0 和 1 的最长连续子数组。
示例 2:
输入: nums = [0,1,0]
输出: 2
说明: [0, 1] (或 [1, 0]) 是具有相同数量0和1的最长连续子数组。
解题
class Solution {
public int findMaxLength(int[] nums) {
int ans = 0,sum = 0;
for(int i=0;iif(nums[i]==0)nums[i]=-1;
HashMapmap = new HashMap();
for(int i=0;isum+=nums[i];
if(sum == 0) { if(i >= ans) ans = i+1; }
if(map.get(sum) == null) { map.put(sum,i); continue; }
int temp = i - map.get(sum);
if(temp > ans) ans=temp;
}
return ans;
}
}