力扣刷经典题
https://leetcode.cn/problems/QTMn0o/submissions/
对比了python和C++的解法。思路完全一样,但是python有个方便的字典get函数,C++使用 map 来当字典用,问题来了。。。。。。
class Solution:
def subarraySum(self, nums: List[int], target: int) -> int:
presum = 0 # 前缀和
dic = {0:1} # dic[key]:value,代表前缀和key出现的次数
ans = 0
# for (auto& x:nums) 只进行一次遍历
for num in nums:
# 1.完成 对前缀串的求和
presum += num
# presum - target 表示以起始点到当前num为终点的前缀和比target多出的距离
# 2.对于当前的求和pre 判断一下是否 前面有之前的 前缀串和 满足 等于 pre-k;
# 3.在python中可以使用字典方便查找结果
ans += dic.get(presum-target,0)
dic[presum] = dic.get(presum,0) + 1
return ans
class Solution {
public:
int subarraySum(vector<int>& nums, int k) {
unordered_map<int, int> mp;
mp[0] = 1;
//count是用来输出的统计次数,pre是对前缀串的求和结果,用mp[pre]++ 来放入到哈希表中,并同时更新次数
int count = 0, pre = 0;
// for (auto& x:nums) 只进行一次遍历
for (auto& x:nums) {
//1.完成 对前缀串的求和
pre += x;
//2.对于当前的求和pre 判断一下是否 前面有之前的 前缀串和 满足 等于 pre-k;
//如果有,我们的 count 需要增加 mp.find(pre - k) 找到的次数
// 判断条件 mp.find(pre - k) != mp.end()
//把 unordered_map 当做字典用
//if (mp.find(pre - k) != mp.end()) {
// count += mp[pre - k];
//}
//加上if可以使程序更加严谨。
if (mp.find(pre - k) != mp.end())
{
count += mp[pre - k];
}
//3.更新存储前缀串求和 的 hash表
mp[pre]++;
}
return count;
}
};
C++ map注意事项
1、在map中,由key查找value时,首先要判断map中是否包含key。
2、如果不检查,直接返回map[key],可能会出现意想不到的行为。如果map包含key,没有问题,如果map不包含key,使用下标有一个危险的副作用,会在map中插入一个key的元素,value取默认值,返回value。也就是说,map[key]不可能返回null。
3、map提供了两种方式,查看是否包含key,m.count(key),m.find(key)。
4、m.count(key):由于map不包含重复的key,因此m.count(key)取值为0,或者1,表示是否包含。
5、m.find(key):返回迭代器,判断是否存在。
6、对于下面的场景,存在key就使用,否则返回null,有下面两种写法:
1 if(m.count(key)>0)
2 {
3 return m[key];
4 }
5 return null;
1 iter = m.find(key);
2 if(iter!=m.end())
3 {
4 return iter->second;
5 }
6 return null;
这里需要注意:前一种方法很直观,但是效率差很多。因为前面的方法,需要执行两次查找。因此,推荐使用后一种方法。
7、对于STL中的容器,有泛型算法find(begin,end,target)查找目标,map还提供了一个成员方法find(key)