一.相关链接
题目链接:454. 四数相加 II
二.心得体会
这道题是使用std::unordered_map的一道题。原因一是不需要排序,但需要元素唯一。原因二是需要快速找到已经存在的值且要求得到这个值的个数,因此需要<key,value>结构。
本题的思路其实很清晰,我们将四个数组分为两个部分。第一个部分两个数组所有和的情况直接存进去map当中作为索引(key),并记录出现次数。然后第二部分就直接查找有没有索引的相反数,有的话取出对应的value加到返回值上即可。
三.代码
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
std::unordered_map<int,int> map12;
int ans=0;
for(int a : nums1){
for(int b : nums2){
//记录每种和的情况出现的次数
map12[a+b]++;
}
}
for(int c : nums3){
for(int d : nums4){
//找到key的相反数
auto iter = map12.find(0-(c+d));
if(iter!=map12.end()){
ans += map12[0-(c+d)];
}
}
}
return ans;
}
};