560. 和为 K 的子数组

【题目】:560. 和为 K 的子数组

方法1. 前缀和

cpp 复制代码
class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int res = 0;
        int n = nums.size();
        vector<int> preSum(n + 1, 0); // 下标从1开始存储
        for(int i = 0; i < n; ++i) {
            preSum[i + 1] = preSum[i] + nums[i];
        }
        for(int i = 0; i < n; ++i) {
            for(int j = i + 1; j <= n; ++j) {
                if(preSum[j] - preSum[i] == k) { // i 到 j 的和
                    ++res;
                }
            }
        }
        return res;
    }
};
  • 时间复杂度: O(n^2)
  • 空间复杂度: O(n)

使用preSum数组来存储nums的前缀和,preSum[i] = nums[0] + nums[1] + ... + nums[i - 1],这里一定要注意,存储的时候必须要从preSum[1]开始存,因为后边计算前缀和的时候,利用的是preSum[j] - preSum[i],得从preSum[1]开始存相减的时候才能不丢失num[i]的值。

方法2. 前缀和 + 哈希表

cpp 复制代码
class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int res = 0;
        unordered_map<int, int> preSumCnt; // 前缀和为key的个数有value个
        preSumCnt[0] = 1; // 对于下标为0的元素,前缀和为0,个数为1
        int cur = 0;
        for(int i = 0; i < nums.size(); ++i) {
            cur += nums[i];
            // 统计cur - k的个数
            if(preSumCnt.find(cur - k) != preSumCnt.end()) {
                res += preSumCnt[cur - k];
            }
            preSumCnt[cur]++;
        }
        return res;
    }
};

使用preSumCnt的map计算前缀和为key的个数有value个。

对于下标为0之前的元素,前缀和为0,个数有1个,这个需要赋初值,后边相加的时候都需要以这个开始。

在遍历的时候,如果找到一个前缀和为cur - k时,说明从之前的某个位置到当前位置的子数组和为k。如果遍历到i时,前缀和为cur,而题目要求找到和为k的子数组。如果存在一个位置j(j < i),其前缀和为cur - k,那么从位置j到位置i的子数组和为k。

相关推荐
朝朝又沐沐3 小时前
算法竞赛阶段二-数据结构(36)数据结构双向链表模拟实现
开发语言·数据结构·c++·算法·链表
薰衣草23333 小时前
一天两道力扣(6)
算法·leetcode
剪一朵云爱着4 小时前
力扣946. 验证栈序列
算法·
遇见尚硅谷4 小时前
C语言:*p++与p++有何区别
c语言·开发语言·笔记·学习·算法
天天开心(∩_∩)4 小时前
代码随想录算法训练营第三十二天
算法
YouQian7724 小时前
(AC)缓存系统
算法·缓存
艾莉丝努力练剑4 小时前
【数据结构与算法】数据结构初阶:详解排序(二)——交换排序中的快速排序
c语言·开发语言·数据结构·学习·算法·链表·排序算法
科大饭桶5 小时前
数据结构自学Day13 -- 快速排序--“前后指针法”
数据结构·算法·leetcode·排序算法·c
李永奉5 小时前
C语言-流程控制语句:for循环语句、while和do…while循环语句;
c语言·开发语言·c++·算法
程序员-King.5 小时前
day69—动态规划—爬楼梯(LeetCode-70)
算法·动态规划