560. 和为 K 的子数组

最新推荐文章于 2024-10-08 10:36:00 发布

ZXY-YXZ

最新推荐文章于 2024-10-08 10:36:00 发布

阅读量67

点赞数

分类专栏：实习刷题文章标签： leetcode 算法

本文链接：https://blog.csdn.net/weixin_47791283/article/details/122516070

版权

实习刷题专栏收录该内容

177 篇文章 0 订阅

订阅专栏

560. 和为 K 的子数组 - 力扣（LeetCode） (leetcode-cn.com)

这题暴力解法超时。。。。(将（0，i）所有的情况列一遍，j=i,j>=0,j--)

优化枚举（0，i）所有子集情况，可以用差来看，(0,i)的和可以拆成（0，j）+（j,i），如果sum(j,i)=k,那统计sum(j,i)=k的次数与统计sum(0，j)出现的次数相同，而sum(0,j)=sum(0,i)-k，这样可以统一到统计从0开始累加和sum(i),出现的次数即可。

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {

        unordered_map<int,int>mp;
        int res=0;
        int sum=0;
        mp[0]=1;
        for(int i=0;i<nums.size();i++){
            sum+=nums[i];
            if(mp.find(sum-k)!=mp.end())
            res+=mp[sum-k];
            mp[sum]++;
        }
        return res;
    }
};