LeetCode454-四数相加II
在一个集合里面判断元素有没有出现过首先想到哈希法
思路
4个数组,元素可以重复使用,最简单暴力的想法就是4个for循环,这样时间复杂度是O(n^4),使用哈希表可以先
遍历前两个列表,将前两个列表遍历后可能出现的结果作统计,存入哈希表中,然后再遍历后两个数组,对后两个数
组中的每一个组合都在哈希表中寻找target=-(nums3[i]+nums4[j])因为题目要求相加为0,查看target在哈希
表中出现的次数,统计所有次数相加即可,这样相比于暴力解法时间复杂度为O(n^2)
代码示例
代码如下(示例):
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int,int> table;
for(int i=0;i<nums1.size();i++){
for(int j=0;j<nums2.size();j++){
table[nums1[i]+nums2[j]]++;
}
}
int count=0;
for(int i=0;i<nums3.size();i++){
for(int j=0;j<nums4.size();j++){
int target=nums3[i]+nums4[j];
auto it=table.find(-target);
if(it!=table.end()){
count+=it->second;
}
}
}
return count;
}