Leetcode 560. 和为K的子数组

最新推荐文章于 2023-10-26 00:15:00 发布

无名小卒一枚

最新推荐文章于 2023-10-26 00:15:00 发布

阅读量112

点赞数

分类专栏： LeetCode 文章标签： C++ 哈希表前缀和数组

本文链接：https://blog.csdn.net/sdau_clt/article/details/115425602

版权

LeetCode 专栏收录该内容

210 篇文章 0 订阅

订阅专栏

该博客讨论了LeetCode的一道题目，要求找到整数数组中和为给定值k的连续子数组的个数。文中提到错误的滑动窗口方法并解释了原因，因为数组元素可能为负数，导致区间和变化的不确定性。接着，文章提出了使用哈希表和前缀和的解决方案，虽然这种方法在处理某些极端情况下可能超时，但其时间复杂度为O(n)，空间复杂度为O(1)。

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

题目描述
给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :

数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-sum-equals-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

C++
错误做法，滑动窗口：
为什么这题不可以用双指针/滑动窗口：得考虑复数。因为nums[i]可以小于0，也就是说右指针i向后移1位不能保证区间会增大，左指针j向后移1位也不能保证区间和会减小。给定j，i的位置没有二段性

方法一：枚举，超时，题解啊。。也顶不住n个1的测试

时间复杂度：O(n^2)
空间复杂度：O(1)。只需要常数空间存放若干变量。

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

哈希表+前缀和

class Solution {
    
    /*哈希表+前缀和：我们定义
    pre[i]为0-i的所有元素的和，pre[i]=pre[i-1]+nums[i]
    可转化为[j,...,k]的子数组和为k:
                pre[i]-pre[j-1]==k;  遍历时对于每一个i,判断这样的j是否存在
                pre[j-1]==pre[i]-k;
    */
public:
    int subarraySum(vector<int>& nums, int k) {
        int res=0;
        int pre=0;
        unordered_map<int,int> thismap;
        thismap[0]=1;
        for(int i=0;i<nums.size();i++)
        {   pre+=nums[i];
            res+=thismap[pre-k];
            thismap[pre]++;
        }
        return res;
    }
};