560. 和为 K 的子数组

560. 和为 K 的子数组

拿到题,一看到数组中和为k,我就想到了用前缀和来做。定义一个前缀和函数perSum,长度为num+1。perSum[i],表示sum[0]sum[i-1]这些值的和。那么很容易就知道sum[i]sum[j]这个区间的和为perSum[j+1]-perSum[i],直接看代码。

class Solution {
    public int subarraySum(int[] nums, int k) {
        //用来返回值
        int res = 0;
        //边界值
        if(nums.length == 0){
            return res;
        }
        int[] perSum = new int[nums.length+1];
        // 构造前缀和
        for(int i = 1; i< perSum.length; i++){
            perSum[i] = perSum[i-1] + nums[i-1];
        }
        for(int i=0 ; i < perSum.length; ++i){
            for(int j=i+1 ; j < perSum.length ; ++j){
                if(perSum[j]-perSum[i]==k){
                    res++;
                }
            }
        }
        return res;
        
    }
}

前缀和也是一种做题技巧。

posted @ 2022-07-14 22:38  抗争的小青年  阅读(28)  评论(0编辑  收藏  举报