做这道题想到的解法是,由于该题只统计元组个数,而不需要位置,那我们首先用一个map来统计数组1和数组2两两元素之和出现的个数,然后去遍历数组3和数组4,在map中寻找 0 - c - d 这个键值是否存在(c和d为数组3和数组4中的元素,如果存在,那么便将元组个数加键值映射的value个。
- 在第一次写时,当数组3和数组4存在这样的元素时,元组个数我写的 ++,而不是+value,导致统计到的个数出现问题。
代码如下:
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map<int, int> mpSumAB;
for(auto a : nums1){
for(auto b : nums2){
mpSumAB[a + b]++;
}
}
int count = 0;
for(auto c : nums3){
for(auto d : nums4){
auto it = mpSumAB.find(0 - c - d);
if(it != mpSumAB.end()){
count += it->second;
}
}
}
return count;
}
};