题目描述:
给你四个整数数组 nums1、nums2、nums3 和 nums4 ,数组长度都是 n ,请你计算有多少个元组 (i, j, k, l) 能满足:
0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0
题目要求将这四个独立数组满足nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0 ,首先可以两两分组,定义一个unordered_map,key作为两数之和,value作为两数之和的次数。首先对前两个数组进行遍历,将两数之和以及它出现的次数放入unordered_map中,然后遍历后两个数组,如果发现nums3和nums4两个数组之和的相反数出现在map中,使用变量count来计算出现的次数。代码如下
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
unordered_map <int,int> map;
int count = 0;
for (int a : nums1) {
for (int b : nums2) {
map[a + b]++;
}
}
for (int a : nums3) {
for (int b : nums4) {
if (map.find(- a - b) != map.end()) {
count += map[- a - b];
}
}
}
return count;
}
};