题目:
给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ,使得 A[i] + B[j] + C[k] + D[l] = 0。
class Solution {
public:
int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
// 定义一个无重复、无序、查找效率高的容器,key为a+b之和,value为出现次数
// 如何确定key和value呢?
// 先要明确自己需要根据谁来查找谁,这个关系搞明白自然就清楚了
unordered_map<int,int> numMap;
int count=0; // 统计四个数相加为01的次数;
for(int a:nums1)
{
for(int b:nums2)
{
/*
如果a+b匹配容器中某个元素的键,函数将返回对其映射值的引用。
如果a+b与容器中任何元素的键都不匹配,函数就会用该键插入一个新元素,并返回对该元素映射值的引用。
请注意,即使没有给元素分配映射值(元素是使用默认构造函数构造的),这也总是将容器大小增加1。
*/
numMap[a+b]++;
}
}
for(int c:nums3)
{
for(int d:nums4)
{
// 肯定是要先判断是否等于0,若等于01,则把count加1
if(numMap.find(0-c-d)!=numMap.end())
count+=numMap[0-c-d];
}
}
return count;
}
};