[力扣 Hot100]Day10 和为 K 的子数组

题目描述

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

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

思路

把前n项和存在数组中,核心公式就是sum[i,j]=sum[0,j]-sum[0,i]+nums[i]

代码

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int n=nums.size();
        int res = 0;
        vector<int> sum(n);
        sum[0]=nums[0];
        for(int i=1; i<n; i++){
            sum[i]=sum[i-1]+nums[i];
        }
        for(int i=0; i<n; i++){
            for(int j=i; j<n; j++){
                if(sum[j]-sum[i]+nums[i]==k)
                    res++;
            }
        }
        return res;
    }
};

你可能感兴趣的:(力扣Hot100题,leetcode,算法,数据结构)