DAY7
454.四数相加II
特点:
1.只用返回元组的个数,而不用返回具体的元组
2.可以不用去重
暴力思路:遍历,这样时间复杂度会达到O(n^4)
标准思路:用哈希法(场景:在一个集合里面判断一个元素是否出现过)
先遍历数组A、B,统计a+b的结果以及这个结果出现的次数,然后遍历数组C、D,判断0-(c+d)有没有在a+b的集合里面出现过。
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int, int> umap; //<key, value>分别代表a+b的值,和这个值出现的次数
for(int a : nums1)
for(int b : nums2)
umap[a + b] ++;
//当访问一个不存在的键时,map 会自动为这个键创建一个新条目
//新条目的值会被默认初始化,对于整型(int),默认初始化的值就是 0
//所以不用去考虑给umap[a + b]设置一个初始值
int count = 0; //最后的结果
for(int c : nums3)
for(int d :nums4)
if(umap.find(0 - (c + d)) != umap.end())
count += umap[0 - (c + d)];
return count;
}
};