写点什么

leetcode 560. Subarray Sum Equals K 和为 K 的子数组 (中等)

作者:okokabcd
  • 2022 年 8 月 21 日
    山东
  • 本文字数:511 字

    阅读完需:约 2 分钟

leetcode 560. Subarray Sum Equals K 和为 K 的子数组(中等)

一、题目大意

https://leetcode.cn/problems/subarray-sum-equals-k


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


示例 1:


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


示例 2:


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


提示:


  • 1 <= nums.length <= 2 * 104

  • -1000 <= nums[i] <= 1000

  • -107 <= k <= 107

二、解题思路

三个思路,第一个:三层遍历,遍历 i 从 0 到 n,第二层遍历 j 从 i 到 n,第三层遍历 i-j,求和,这种方法会超时


第二个:求累加和 sum[i] = num[0]-num[i-1],这样二层遍历得到(i,j),检查 sum[j]-sum[i]


第三个:定义一个 hash 来保存 sum 出现的次数,这样累加 sum-k 出现的次数就是答案了

三、解题方法

3.1 Java 实现

class Solution {    public int subarraySum(int[] nums, int k) {        int ans = 0;        int sum = 0;        Map<Integer, Integer> d = new HashMap<>();        d.put(0, 1);        for (int i = 0; i < nums.length; i++) {            sum += nums[i];            ans += d.getOrDefault(sum - k, 0);            d.put(sum, d.getOrDefault(sum, 0) + 1);        }        return ans;    }}
复制代码

四、总结小记

  • 2022/8/21 得买个头盔了

发布于: 刚刚阅读数: 5
用户头像

okokabcd

关注

还未添加个人签名 2019.11.15 加入

一年当十年用的Java程序员

评论

发布
暂无评论
leetcode 560. Subarray Sum Equals K 和为 K 的子数组(中等)_LeetCode_okokabcd_InfoQ写作社区