【CODE】Subarray Sum Equals K

560. Subarray Sum Equals K

Medium

283976FavoriteShare

Given an array of integers and an integer k, you need to find the total number of continuous subarrays whose sum equals to k.

Example 1:

Input:nums = [1,1,1], k = 2
Output: 2

Note:

  1. The length of the array is in range [1, 20,000].
  2. The range of numbers in the array is [-1000, 1000] and the range of the integer k is [-1e7, 1e7].
class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int res=0,n=nums.size();
        for(int i=0;i<n;i++){
            int sum=nums[i];
            if(sum==k) res++;
            for(int j=i+1;j<n;j++){
                sum+=nums[j];
                if(sum==k) res++;
            }
        }
        return res;
    }
};
class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int res=0,n=nums.size(),sum=0;
        unordered_map<int,int> m{{0,1}};
        for(int i=0;i<n;i++){
            sum+=nums[i];
            res+=m[sum-k];
            m[sum]++;
        }
        return res;
    }
};
发布了133 篇原创文章 · 获赞 35 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/Li_Jiaqian/article/details/103491067
今日推荐