每日一道算法面试题(16):leecode 560 和为K的子数组

1.算法题目

给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。

说明 :

数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ,且整数 k 的范围是 [-1e7, 1e7]。

2.算法思路

  1. 暴力法:考虑给定数组 sums 的每个可能的子数组,找到每个子数组的总和,并检查总和是否等于 k,等于时递增用于记录所需结果的 count;时间复杂度为 O(n^3),空间复杂度为 O(1),超出时间限制;
  2. 使用累计和:计算数组的每个数字开始到累加后面每个数字的和,遇到和为 k时, count 加1;需要两层循环遍历,时间复杂度为 O(n^2),空间复杂度为 O(1);
  3. 使用哈希表:使用一个哈希表 map ,它用于存储所有可能的索引的累积总和以及相同累加和发生的次数,遍历数组 nums 并继续寻找累积总和。每当我们遇到一个新的和时,我们在hashmap中创建一个与该总和相对应的新条目。如果再次出现相同的和,我们增加与 map 中的和相对应的计数;对于遇到的每个总和,确定已经发生 sum-k 总和的次数,因为它将确定具有总和 k 的子阵列到当前索引的次数,同时将 count 增加相同的数量。最后得到的 count 即为满足题意的值。只需要一次循环遍历数组,但需要 map的辅助空间,时间复杂度为 O(n),空间复杂度为 O(n);

3.算法代码

根据累计和算法思路,编写的算法代码如下:

public static int subarraySum(int[] nums, int k) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        int count = 0;
        for (int i = 0; i < nums.length; i ++) {
            int sum = 0;
            for (int j = i; j < nums.length; j ++) {
                sum += nums[j];
                if (sum == k) {
                    count ++;
                }
            }
        }
        return count;
    }

根据哈希表算法思路,编写的算法代码如下:

    public int subarraySum(int[] nums, int k) {
        if (nums == null || nums.length == 0) {
            return 0;
        }
        HashMap<Integer, Integer> map = new HashMap<>();
        int count = 0, sum = 0;
        map.put(0, 1);
        for (int i = 0; i < nums.length; i ++) {
            sum += nums[i];
            if (map.containsKey(sum - k)) { // 和为 sum - k 的数据与和为 sum 的区间数的和即为 k,这个区间的数即为满足题意的和为 k 的连续的子数组的数
                count += map.get(sum - k);
            }
            // 保存和为 sum 的数量
            map.put(sum, map.getOrDefault(sum, 0) + 1);
        }
        return count;
    }

如果你有疑问或更好的算法思路,欢迎留言交流!!!

©️2020 CSDN 皮肤主题: 大白 设计师:CSDN官方博客 返回首页