454.四数相加

记录新手小白的做题过程。

题目：

给你四个整数数组 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

---

 思路：乍一看感觉题很难，主要是脑子里只闪过暴力解法，先存好第一个数，在往第二个数组里找第三个数，然后第三个，第四个，这样的时间复杂度是O(n^2)。

---

看一下别人的代码：

点击链接

本题解题步骤：

1. 首先定义 一个unordered_map，key放a和b两数之和，value 放a和b两数之和出现的次数。//a,b两数之和出现的次数

2. 遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中。

3. 定义int变量count，用来统计a+b+c+d = 0 出现的次数。

4. 再遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就用count把map中key对应的value也就是出现次数统计出来。

5. 最后返回统计值 count 就可以了

他的思路就是将数组分成两拨，分别计算和以及这个和出现的次数，因为是map，不允许key重复，所以不用担心重复，然后将问题再简化为两个数的和

代码如下：

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        unordered_map<int, int> umap; //key:a+b的数值，value:a+b数值出现的次数
        // 遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中
        for (int a : A) {//一种遍历方式
            for (int b : B) {
                umap[a + b]++;//哇哦这种记录方式很棒欸！学到了
            }
        }
        int count = 0; // 统计a+b+c+d = 0 出现的次数
        // 在遍历大C和大D数组，找到如果 0-(c+d) 在map中出现过的话，就把map中key对应的value也就是出现次数统计出来。
        for (int c : C) {
            for (int d : D) {
                if (umap.find(0 - (c + d)) != umap.end()) {//第二个就简单一点，只要记录能够合成0的就可以了，并记载次数
                    count += umap[0 - (c + d)];//每找到一个，就可以与符合要求的所有数字组合
                }
            }
        }
        return count;
    }
};

---

收获：

1.这种记录和的方式真的特别棒欸，学到了！

  // 遍历大A和大B数组，统计两个数组元素之和，和出现的次数，放到map中
        for (int a : A) {//一种遍历方式
            for (int b : B) {
                umap[a + b]++;
            }
        } 

---

感觉不错，下一题！