[leetcode] 560. Subarray Sum Equals K

最新推荐文章于 2024-08-24 08:32:38 发布

TstsUgeg

最新推荐文章于 2024-08-24 08:32:38 发布

阅读量4.7k

点赞数 1

分类专栏： leetcode 文章标签： Hash Table

本文链接：https://blog.csdn.net/TstsUgeg/article/details/71107676

版权

leetcode 专栏收录该内容

254 篇文章 0 订阅

订阅专栏

给定一个整数数组和整数k，求和为k的连续子数组的总数。使用哈希表记录前缀和，以此优化双层循环遍历，提高算法效率。

摘要由CSDN通过智能技术生成

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:

The length of the array is in range [1, 20,000].
The range of numbers in the array is [-1000, 1000] and the range of the integer k is [-1e7, 1e7].

这道题是统计相加和为k的连续子数组个数，题目难度为Medium。

题目和第523题有些类似，感兴趣的同学可以顺便看下第523题（传送门）。

最直观的想法是遍历数组并依次加当前位置的数字，同时用数组sum记录下当前位置之前所有数字的相加和，这样下标[i, j)之间的数字之和就可以用sum[j]-sum[i]来计算，然后通过双层循环，遍历所有情况来统计满足条件的子数组个数。具体代码：

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

用sum表示从数组开始位置到当前位置的数字相加和，有了第523题的经验，我们还可以用Hash Table来存储sum出现的次数，如果当前位置之前有相加和为sum-k的位置，则这两个位置之间的数字相加和为k，以当前位置结尾的相加和为k的子数组个数为hash[sum-k]，这样遍历整个数组即可得出满足条件的子数组个数。具体代码：

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int sum = 0, cnt = 0;
        unordered_map<int, int> hash;
        
        hash[0] = 1;
        for(auto n:nums) {
            sum += n;
            cnt += hash[sum-k];
            ++hash[sum];
        }
        
        return cnt;
    }
};