838. Subarray Sum Equals K

最新推荐文章于 2020-04-13 08:29:08 发布

Vest-Lee

最新推荐文章于 2020-04-13 08:29:08 发布

阅读量158

点赞数

分类专栏：刷题

本文链接：https://blog.csdn.net/vestlee/article/details/80720904

版权

刷题专栏收录该内容

215 篇文章 1 订阅

订阅专栏

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

  您在真实的面试中是否遇到过这个题？   
 是 

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

基本思路都差不多，就是建立累加数组。使用关联列表会更简单一点。

class Solution {
public:
    /**
     * @param nums: a list of integer
     * @param k: an integer
     * @return: return an integer, denote the number of continuous subarrays whose sum equals to k
     */
    int subarraySumEqualsK(vector<int> &nums, int k) {
        // write your code here
        int res = 0, sum = 0, n = nums.size();
        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;
    }
};