Day7 四数相加II

力扣题目链接(opens new window)

给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ，使得 A[i] + B[j] + C[k] + D[l] = 0。

为了使问题简单化，所有的 A, B, C, D 具有相同的长度 N，且 0 ≤ N ≤ 500 。所有整数的范围在 -2^28 到 2^28 - 1 之间，最终结果不会超过 2^31 - 1 。

例如:

输入:

• A = [ 1, 2]
• B = [-2,-1]
• C = [-1, 2]
• D = [ 0, 2]

输出:

2

解释:

两个元组如下:

1. (0, 0, 0, 1) -> A[0] + B[0] + C[0] + D[1] = 1 + (-2) + (-1) + 2 = 0
2. (1, 1, 0, 0) -> A[1] + B[1] + C[0] + D[0] = 2 + (-1) + (-1) + 0 = 0 

本题解题步骤：

1. 首先定义 一个unordered_map，key放a和b两数之和，value 放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 就可以了

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        multiset<int> mulset1;
        unordered_map<int, int> map1;
        for (int num1 : nums1) {
            for (int num2 : nums2) {
                //先遍历nums1和nums2，将他们的数值相加，统计每个值出现的次数，并添加到map中：
                //key:两个数组中元素相加的和；value对应这个和出现的次数：
                //这是map中插入数据的一种方式：m1[4] = 40; 4对应key，value对应40；
                map1[num1 + num2]++;//这句话应该重点理解；
                // mulset1.insert(num1+num2);
            }


        }

        int count = 0;
        for (int num3 : nums3) {
            for (int num4 : nums4) {
                int res1 = num3 + num4;
                //在map1中找加上res1等于0的那些key，然后将他们的value叠加起来
                int res2 = 0 - res1;
                unordered_map<int, int>::iterator pos = map1.find(res2);
                if (pos != map1.end()) {
                    //将找到的res2对应的value叠加到count中，这里的value值是res2出现的次数
                    //map1[res2]的这种方式可以非常方便获取到res2对应的value值是什么；
                    count += map1[res2];
                }



            }


        }
        return count;
    }
};