LeetCode454 四数相加Ⅱ

题目： 给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

• 0 <= i, j, k, l < n
• nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

示例 1：

输入：nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出：2

示例 2：

输入：nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
输出：1

思路：

四数相加等于0，这里使用哈希map集合来存放，首先将A和B集合中的每种类型相加，map的键是A+B的值，值是A+B等于某个数的个数；再从C+D中找[0-(c+d)]，找到的话，给count加上A+B中这个数的个数，也就是[0-(c+d)]。使用两个两层for循环做的时间复杂度最小。

class Solution {
public:
	int foursum(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D ) {
		unordered_map<int, int> map;
		for (int a:A) {
			for (int b:B) {
				map[a + b]++;
			}
		}
		int count = 0;
		for (int c:C) {
			for (int d:D) {
				if (map.find(0 - (c + d))!=map.end()) {
					count += map[0-(c+d)];
				}
			}
		}
		return count;
	}
};

int main() {
	vector<int> A = { 1,2 };
	vector<int> B = { -2,-1};
	vector<int> C = { -1,2 };
	vector<int> D = { 0,2 };
	Solution ss;
	cout << ss.foursum(A, B, C, D) << endl;
	return 0;
}