【前缀和】560. 和为 K 的子数组

最新推荐文章于 2024-07-19 22:36:19 发布

少写代码少看论文多多睡觉

最新推荐文章于 2024-07-19 22:36:19 发布

阅读量200

点赞数 2

分类专栏： # Leetcode 文章标签：算法数据结构

本文链接：https://blog.csdn.net/qq_44653420/article/details/136251077

版权

Leetcode 专栏收录该内容

487 篇文章 1 订阅

订阅专栏

该篇文章介绍了一种使用前缀和数组的方法来求解给定整数数组中和为K的子数组个数。作者提供了Solution类的代码，通过遍历并计算前缀和，找出满足条件的子数组并计数。

摘要由CSDN通过智能技术生成

560. 和为 K 的子数组

解题思路

创建一个前缀和数组 preSum，其长度比原数组 nums 多 1。preSum[i] 表示 nums 中前 i 个元素的和。
通过遍历 nums 数组，计算前缀和数组 preSum。
- 在嵌套的两个循环中，对所有可能的子数组进行穷举：
- 外层循环从 1 到 n（n 是数组长度），代表子数组的结束位置。
- 内层循环从 0 到 i-1，代表子数组的起始位置。
在每一对起始位置和结束位置上，通过计算前缀和数组中的差值，判断子数组的和是否等于 k。
如果等于 k，则将结果计数器 ans 加一。

class Solution {
    public int subarraySum(int[] nums, int k) {
        // 前缀和数组
        int n = nums.length;
        int[] preSum = new int[n + 1]; 
        preSum[0] = 0;

        for(int i = 0; i < n; i++){
            preSum[i + 1] = preSum[i] + nums[i];
        }

        // preSum[i] 代表nums[0...i - 1]的和

        int ans = 0;
        // 穷举所有子数组
        for(int i = 1; i <= n; i++){
            for(int j = 0; j < i; j++){
                if(preSum[i] - preSum[j] == k){
                    ans++;
                }
            }
        }


        return ans;
    }
}