1、哈希表
我们可以使用哈希表来加速查找。考虑到数组中存在四个元素,我们可以将元素两两拆分,先将数组nums1、nums2中的元素之和加入哈希表中并统计次数,而后在哈希表中查找是否存在nums3 和 nums4元素之和的相反值,若存在则在结果中加入当前出现的次数,否则直接跳过。
class Solution {
public:
int fourSumCount(vector<int> &nums1, vector<int> &nums2, vector<int> &nums3, vector<int> &nums4) {
int res = 0;
unordered_map<int, int> hs;
for (int i: nums1) {
for (int j: nums2) {
++hs[i + j];
}
}
for (int i: nums3) {
for (int j: nums4) {
if (hs.count(-i - j)) {
res += hs[-i - j];
}
}
}
return res;
}
};