【前缀和】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;
    }
}
相关推荐
AAAA劝导tx8 分钟前
List--链表
数据结构·c++·笔记·链表·list
格格Code8 分钟前
八大排序——冒泡排序/归并排序
数据结构·算法·排序算法
C语言魔术师39 分钟前
62.不同路径
算法·leetcode·动态规划
fantasy_440 分钟前
LeetCode238☞除自身以外数组的乘积
java·数据结构·python·算法·leetcode
椰羊~王小美1 小时前
贪心算法和动态规划
算法·贪心算法·动态规划
啊阿狸不会拉杆2 小时前
人工智能数学基础(二):初等数学
人工智能·python·算法
元亓亓亓2 小时前
LeetCode热题100--560.和为K的子数组(前缀和)--中等
算法·leetcode·职场和发展
Phoebe鑫2 小时前
数据结构每日一题day12(链表)★★★★★
数据结构·算法·链表
Vacant Seat2 小时前
贪心算法-跳跃游戏II
算法·游戏·贪心算法
夜松云3 小时前
从对数变换到深度框架:逻辑回归与交叉熵的数学原理及PyTorch实战
pytorch·算法·逻辑回归·梯度下降·交叉熵·对数变换·sigmoid函数