刷题12-和为k的子数组

刷题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;
    }
}