2021/01/12打卡

最新推荐文章于 2025-11-24 01:23:29 发布

原创最新推荐文章于 2025-11-24 01:23:29 发布 · 173 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode打卡专栏收录该内容

11 篇文章

订阅专栏

博客围绕LeetCode 560题和为K的子数组展开，介绍了解题思路。先提及三重暴力循环，其时间复杂度为O(n^3)会超时，接着是优化后的暴力解法，最后重点讲解利用前缀和加哈希表的方法，通过记录前缀和出现次数来求解。

560 和为K的子数组

在这里插入图片描述
解题思路：

最先向想到的是三重暴力循环：如下代码所示：时间复杂度为O(n^3);超时

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

然后是优化后的暴力解法

int cnt = 0;
        int len = nums.size();
        for(int i=0;i<len;i++){
            int sum=0;
            for(int j = i;j < len;j++){
                sum+=nums[j];
                if(sum==k) cnt++;
            }
        }
        return cnt;

最后是利用前缀和加上哈希表
求j到i的和为k的子数组，等价于求到i的前缀和为pre-k的数组个数；对数组遍历一遍，记录每个前缀和出现的次数，从前往后，如果出现多次，每次累加一即可；

 unordered_map<int,int>mp;
        //和为键，出现次数为值
        mp[0]=1;
        int cnt=0;
        int pre=0;
        for(auto num:nums){
            pre+=num;
            if(mp.find(pre-k)!=mp.end()){
                cnt+=mp[pre-k];
            }
            mp[pre]++;;
        }
        return cnt;