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

560. 和为 K 的子数组

解题思路

  • 创建一个前缀和数组 preSum,其长度比原数组 nums 多 1。preSum[i] 表示 nums 中前 i 个元素的和。
  • 通过遍历 nums 数组,计算前缀和数组 preSum。
    • 在嵌套的两个循环中,对所有可能的子数组进行穷举:
    • 外层循环从 1 到 n(n 是数组长度),代表子数组的结束位置。
    • 内层循环从 0 到 i-1,代表子数组的起始位置。
  • 在每一对起始位置和结束位置上,通过计算前缀和数组中的差值,判断子数组的和是否等于 k。
  • 如果等于 k,则将结果计数器 ans 加一。
java 复制代码
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;
    }
}
相关推荐
HR Zhou32 分钟前
群体智能优化算法-算术优化算法(Arithmetic Optimization Algorithm, AOA,含Matlab源代码)
人工智能·算法·数学建模·matlab·优化·智能优化算法
wen__xvn33 分钟前
每日一题洛谷P8649 [蓝桥杯 2017 省 B] k 倍区间c++
c++·算法·蓝桥杯
倔强的石头10634 分钟前
【C++经典例题】杨辉三角问题
算法
星星火柴9361 小时前
数据结构:链表 (C++实现)
数据结构·c++·笔记·链表
在努力的韩小豪1 小时前
B树和B+树的区别(B Tree & B+ Tree)
数据结构·数据库·b树·b+树·索引·数据库索引
独好紫罗兰1 小时前
洛谷题单3-P4956 [COCI 2017 2018 #6] Davor-python-流程图重构
开发语言·python·算法
ん贤1 小时前
2024第十五届蓝桥杯大赛软件赛省赛C/C++ 大学 B 组
c语言·数据结构·c++·经验分享·笔记·算法·蓝桥杯
PownShanYu2 小时前
RainbowDash 的 Robot
算法
Phoebe鑫2 小时前
数据结构每日一题day11(链表)★★★★★
数据结构·算法
独好紫罗兰3 小时前
洛谷题单3-P2669 [NOIP 2015 普及组] 金币-python-流程图重构
开发语言·python·算法