力扣560和为K的数组

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

子数组是数组中元素的连续非空序列。

示例 1:

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

示例 2:

输入:nums = [1,2,3], k = 3
输出:2

解法一:暴力

    public int subarraySum(int[] nums, int k) {
        int count = 0;
        //定义左指针
        for (int left = 0; left < nums.length; left++) {
            int sum = 0;
            //右指针
            for (int right = left; right < nums.length; right++) {
               sum += nums[right];
               if(sum > k){
                   break;
               }
               if(sum == k){
                   count++;
               }
            }
        }
        return count;
    }

解法二:前缀和

先计算出前缀和,再左右指针根据前缀和方便计算

    public int subarraySum(int[] nums, int k) {
      int count  = 0;
      //计算前缀和,构造数组
        int[] preSum = new int[nums.length+1];
        //起始元素设置为0,代表
        preSum[0] = 0;
        //初始化前缀和数组
        for (int i = 0; i < nums.length; i++) {
            preSum[i+1] = nums[i] + preSum[i];
        }
        //左指针
        for (int left = 0; left < nums.length; left++) {
            //右指针
            for(int right = left; right < nums.length; right++){
                //利用公式sum=preSum[right+1]−preSum[left]计算区间有无符合数据
                if(preSum[right+1] - preSum[left] == k){
                    count++;
                }
            }
        }

        return count;
    }

解法三:前缀和+哈希

哈希是通过计算两个前缀和的差值是否为K,当为K的时候,count++

    public int subarraySum(int[] nums, int k) {
        int count  = 0;
        HashMap<Integer, Integer> preSumFreq  = new HashMap<>();
        //计算前缀和,构造数组,就是从

        //为了防止开头第一个元素直接就是k,因为我们是先containKey,如果没有初始值,那就会漏一个结果
        preSumFreq.put(0, 1);
        int preSum = 0;
        //初始化前缀和map,键是前缀和,值是该前缀和出现的频率。
        for (int num : nums) {
            preSum += num;
            //判断有该元素是否是符合结果
            if(preSumFreq.containsKey(preSum - k)){
                count += preSumFreq.get(preSum - k);
            }
            preSumFreq.put(preSum, preSumFreq.getOrDefault(preSum, 0) + 1);

        }


        return count;
    }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值