ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

leetcode-560. 和为K的子数组

2021-07-25 14:34:03  阅读:135  来源: 互联网

标签:pre count nums 560 ++ int map 数组 leetcode


 

 

前缀和与哈希表

// class Solution {
// public:
//     int subarraySum(vector<int>& nums, int k) {
//         // 前缀和
//         // 超时

//         vector<int> pre(nums.size()+1,0);
//         pre[0] = 0;
//         for(int i = 0; i < nums.size(); i++){
//             pre[i+1] = pre[i] + nums[i];       
//         }

//         int count = 0;
//         for(int i = 0; i < nums.size(); i++)
//             for(int j = i; j < nums.size(); j++){
//                 if(pre[j+1]-pre[i] == k)
//                     count++;
//             }

//         return count;
//     }
// };


class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        // 前缀和
        // 超时
        unordered_map<int,int> in_map;
        int pre = 0;
        int count = 0;
        in_map[0] = 1;   // 避免了in_map.find(0)找不到的现象
        for(int i = 0; i < nums.size(); i++){
            pre = pre + nums[i];  // 前缀和
            if(in_map.find(pre-k)!=in_map.end()){
                //count = count + 1;
                count = count + in_map[pre-k];
                // 为什么不用上面一个,因为前缀和可能有多个序列等于同一个值。
            }
            in_map[pre]++;
        }
        return count;
    }
};

 

标签:pre,count,nums,560,++,int,map,数组,leetcode
来源: https://www.cnblogs.com/ymec/p/15057744.html

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有