LeetCode-560-和为K的子数组

LeetCode-560-和为K的子数组

思路

很容易想到就是暴力枚举法，但是时间复杂度太高了，这里采用的是前缀和+哈希表的方法。

• 引入前缀和的概念：就是数组前几个元素的和，有以下递推式：pre[i]=pre[i-1]+nums[i]。
• 目标是找到连续元素，和为k，即存下以下关系：pre[i]-pre[j-1]=k
• 使用了哈希表来存储pre，因此只需要维护一个pre，遍历nums数组时更新pre，同时判断是否存在pre[j-1]=pre[i]-k

代码

class Solution {
    public int subarraySum(int[] nums, int k) {
        Map<Integer,Integer> map=new HashMap<>();//存储之前的前缀和
        int pre=0;//只维护一个pre
        int cnt=0;
        map.put(0,1);//保证前缀和和k相等存在
        for(int num:nums){
            pre+=num;//更新pre
            if(map.containsKey(pre-k)){
                cnt+=map.get(pre-k);
            }//判断pre[j-1]=pre[i]-k是否存在
            map.put(pre,map.getOrDefault(pre,0)+1);//放入当前pre到哈希表
        }
        return cnt;
    }
}