其他分享
首页 > 其他分享> > 560. 和为 K 的子数组

560. 和为 K 的子数组

作者:互联网

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;
    }
};

标签:次数,560,sum,int,数组,统计
来源: https://blog.csdn.net/weixin_47791283/article/details/122516070