【LeetCode】和为K的子数组

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

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

示例 1:

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

输出:2

示例 2:

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

输出:2

核心思路:

转化子数组的和问题:

  • 假设我们有两个前缀和 prefix_sum[i] 和 prefix_sum[j](其中 i < j)。那么,nums[i+1] 到 nums[j] 这一段子数组的和就是 prefix_sum[j] - prefix_sum[i]。
  • 如果要使得某一段子数组的和为 k,即 prefix_sum[j] - prefix_sum[i] = k,可以转化为:prefix_sum[j] = prefix_sum[i] + k。
cpp 复制代码
#include <vector>
#include <unordered_map>
#include <iostream>

using namespace std;

class Solution 
{
public:
    int subarraySum(vector<int>& nums, int k) 
    {
        int count = 0;                                  // 记录和为 k 的子数组个数
        int prefix_sum = 0;                             // 前缀和
        unordered_map<int, int> prefix_sum_count;       // 用于存储每个前缀和出现的次数
        prefix_sum_count[0] = 1;                        // 初始时前缀和为0出现一次

        for (int num : nums)
        {
            prefix_sum += num;
            if (prefix_sum_count.find(prefix_sum - k) != prefix_sum_count.end())
            {
                count += prefix_sum_count[prefix_sum - k];
            }
            prefix_sum_count[prefix_sum]++;
        }
        return count;
    }
};

int main() 
{
    Solution solution;
    vector<int> nums1 = { 1, 1, 1 };
    int k1 = 2;
    int result1 = solution.subarraySum(nums1, k1);

    vector<int> nums2 = { 1, 2, 3 };
    int k2 = 3;
    int result2 = solution.subarraySum(nums2, k2);
    return 0;
}
相关推荐
丁浩6663 小时前
Python机器学习---2.算法:逻辑回归
python·算法·机器学习
伏小白白白3 小时前
【论文精度-2】求解车辆路径问题的神经组合优化算法:综合展望(Yubin Xiao,2025)
人工智能·算法·机器学习
无敌最俊朗@4 小时前
数组-力扣hot56-合并区间
数据结构·算法·leetcode
囚生CY4 小时前
【速写】优化的深度与广度(Adam & Moun)
人工智能·python·算法
码农多耕地呗5 小时前
力扣94.二叉树的中序遍历(递归and迭代法)(java)
数据结构·算法·leetcode
微笑尅乐5 小时前
BFS 与 DFS——力扣102.二叉树的层序遍历
leetcode·深度优先·宽度优先
懒羊羊不懒@5 小时前
Java基础语法—最小单位、及注释
java·c语言·开发语言·数据结构·学习·算法
白云千载尽6 小时前
leetcode 912.排序数组
算法·leetcode·职场和发展
哆啦刘小洋6 小时前
Tips:预封装约束的状态定义
算法
代码充电宝6 小时前
LeetCode 算法题【简单】290. 单词规律
java·算法·leetcode·职场和发展·哈希表