刷题12-010-和为k的子数组
核心思想:
需要一个哈希表,key为前i个数字之和,value为每个和出现的次数
class Solution {
public int subarraySum(int[] nums,int k){
Map<Integer,Integer> sumtocount=new HashMap<>();
sumtocount.put(0,1);//key为前i个数字之和,value为每个和出现的次数
int sum=0;
int count=0;
for(int n:nums){
sum+=n;//getOrDefault函数返回指定key对应的value
count+=sumtocount.getOrDefault(sum-k,0);//
sumtocount.put(sum,sumtocount.getOrDefault(sum,0)+1);
}
return count;
}
}
注意是连续的子数组!而且当扫描到数组的第i个数字并求得前i个数字之和是x时,需要知道在i之前存在多少个j且前j个数字之和等于x-k
题目011-0和1个数相同的子数组
这几道题目都是求连续子数组的,要转换一下思维
class Solution {
public int findMaxLength(int[] nums) {
int n = nums.length, sum = 0, res = 0;
HashMap<Integer, Integer> map = new HashMap<>();
/**
前缀和+HashMap
*/
// 和为0的时候的下标是-1, 哨兵点,当起始区间的和是0的时候,用到哨兵
map.put(0, -1);
for (int i = 0; i < n; i ++){
if (nums[i] == 0) sum -= 1;
else sum += 1;
// System.out.println(i + " " + sum);
// 对于区间求和, 如果sum-0 == sum1 存在map中,相当于sum - sum1 = 0
// 那么这一段区间之和就是0
if (map.containsKey(sum)){
res = Math.max(res, i - map.get(sum));
}else{
map.put(sum, i);
}
}
return res;
}
}