LeetCode_数据结构_和为k的子数组

和为k的子数组

给你一个整数数组 nums 和一个整数 k ，请你统计并返回该数组中和为 k 的连续子数组的个数。

示例 1：

输入：nums = [1,1,1], k = 2
输出：2

（上述题目来源于LeetCode）

解法一（暴力解法1）

class Solution {
    public int subarraySum(int[] nums, int k) {
        //暴力解法
        int count = 0;
        int n = nums.length;
        for(int i = 0;i < n;i++){
            int result = 0;
            for(int j = i;j < n;j++){
                result = result + nums[j];
                if(result == k){
                    count++;
                }
            }
        }
        return count;
    }
}

解法二（暴力解法2）

class Solution {
    public int subarraySum(int[] nums, int k) {
        //枚举
        int count = 0;
        int n = nums.length;
        for(int i = 0;i < n;i++){
            int result = 0;
            for(int j = i;j >= 0;j--){
                result = result + nums[j];
                if(result == k){
                    count++;
                }
            }
        }
        return count;
    }
}

解法三（辅助数组）

class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0;
        int n = nums.length;
        int[] sum1 = new int[n];
        int[] sum2 = new int[n];
        sum1[0] = nums[0];
        sum2[0] = nums[0];
        for(int i = 1;i < n;i++){
            sum1[i] = sum1[i - 1] + nums[i];
            sum2[i] = sum2[i - 1] + nums[i];
        }
        for(int i = 0;i < n;i++){
            if(sum1[i] == k){
                    count++;
                }
            for(int j = i + 1;j < n;j++){
                if(sum2[j] - sum1[i] == k){
                    count++;
                }
            }
        }
        return count;
    }
}

解法四（前缀和+哈希表）

class Solution {
    public int subarraySum(int[] nums, int k) {
        //前缀和 + 哈希表
        int count = 0;
        int n = nums.length;
        //将放置前缀和的数组长度置为n+1是因为要和下方新建的哈希表一一对应
        int[] sum = new int[n + 1];
        Map<Integer,Integer>hashmap = new HashMap<>();
        //哈希表的第一位放(0,1)是因为方便找到前缀和等于k的情况
        hashmap.put(0,1);
        //计算前缀和
        for(int i = 0;i < n;i++){
            sum[i + 1] = sum[i] + nums[i];
            //如果哈希表中存在key为sum[i] - k
            if(hashmap.containsKey(sum[i + 1] - k)){
                count = count + hashmap.get(sum[i + 1] - k);
            }
            //将当前的前缀和放入哈希表中
            hashmap.put(sum[i + 1],hashmap.getOrDefault(sum[i + 1],0) + 1);
        }
        return count;
    }
}