C++力扣题目454--四数相加II

给你四个整数数组 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
解释：
两个元组如下：
1. (0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0

示例 2：

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

思路：四个数相加，可以拆分成1个和3个相加或者2，2相加，但3个相加具体实现时就回出现三层循环，时间复杂度过大，因此使用2,2相加，步骤就是创建一个哈希表，key用来存放两个数组内元素相加的结果，value记录两个数组内元素相加结果出现的次数，然后利用哈希表的特点，对另两个数组内元素的和的相反数进行查找，若找到，计数加上该结果的出现的次数即value。代码如下：

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        int n = nums1.size(), count = 0;
        unordered_map<int, int>map1;//相加的结果为key,结果出现的次数为value
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < n; j++)
            {//统计前两个数组内所有元素两两相加的结果及该结果出现的次数
                map1[(nums1[i] + nums2[j])]++;
            }
        }
        for (int i = 0; i < n; i++)
        {
            for (int j = 0; j < n; j++)
            {//将后两个数组所有元素两两相加的结果取反后在map1中查找，若找到,加上对应的次数
                if (map1.find(-(nums3[i] + nums4[j])) != map1.end())
                {
                    count += map1[-(nums3[i] + nums4[j])];
                }
            }
        }
        return count;//返回count，即为最终的答案
    }
};