class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
// nums1 与 nums2 的和作为key,出现的次数作为 value
unordered_map<int, int> sum12;
for (auto i : nums1) {
for (auto j : nums2) {
sum12[i + j]++;
}
}
// 0 - (nums3 + nums4) 在sum12中是否出现,并将次数累加
int count = 0;
for (auto i : nums3) {
for (auto j : nums4) {
if (sum12.find(0 - (i + j)) != sum12.end()) {
count += sum12[0 - (i + j)];
}
}
}
return count;
}
};
总结
- 定义一个map 存储,数组1、2 的{和,次数}
- 在map里查找(0-nums3[i]-nums4[j]),并累加次数