算法力扣刷题记录 十七【454题.四数相加II】

前言

哈希篇，继续。
记录 十七【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

提示：

n == nums1.length
n == nums2.length
n == nums3.length
n == nums4.length
1 <= n <= 200
-228 <= nums1[i], nums2[i], nums3[i], nums4[i] <= 228

---

二、尝试实现

思路

（1）一下子操作4个数组，数量多，怎么分开操作呢？比如两个两个的操作。那么求和等于0，可以想把nums3[k] + nums4[l] 移到等式的右边。这样即使遍历，也是2层循环遍历。

• 第一步：把nums3[k] + nums4[l] 移到等式的右边。（负号问题不大）

（2）等式左边：nums1[i] + nums2[j]求和。等式右边：0 - (nums3[k] + nums4[l] )第二轮求和。那么：把nums1[i] + nums2[j]所有的和放到一个哈希结构中，等到求和nums3+nums4的时候，判断这个和的相反数有没有在哈希结构中出现过。出现过，说明元组数可以增加。没有，说明配对不成功。

• 第二步：确定用哈希结构。开始选哪一种哈希结构。数组？set？map？

（3）目标返回元组数量。假设[0,1,1,1]满足，[1,0,1,1]也满足，这是两个元组：nums3和nums4的下标对一样，说明nums1[0]+nums2[1] = 3和nums1[1]+nums2[0] = 3求和一样，那么和对应的坐标对数量加1，但是和不用重复。

• 容器特点：每一个和，需要记录对应的坐标对数量。和作为key，坐标对数量作为value。所以在map中选。顺序重要吗？能找到代表出现就好，顺序不重要，所以无序。和不用重复，插入不成功时，对应value++即可。
• 确定容器结构是unordered_map 。

代码实现

class Solution {
public:
    int fourSumCount(vector<int>& nums1, vector<int>& nums2, vector<int>& nums3, vector<int>& nums4) {
        unordered_map<int,int> map;//放nums1+nums2所有可能得和。和作为key，坐标对的数量作为value
        int result = 0;	//返回值先摆上
        
        //遍历nums1和nums2
        for(int i=0;i < nums1.size();i++){
            for(int j = 0;j < nums2.size();j++){	
                auto it = map.insert({0-nums1[i]-nums2[j],1});	//相反数在这里处理
                if(it.second == false){ //没有成功插入，多了一个坐标对,数量+！
                    it.first->second ++;
                }
            }
        }

		//遍历nums3和nums4
        for(int i = 0;i < nums3.size();i++){
            for(int j = 0;j < nums4.size();j++){
                auto it = map.find(nums3[i]+nums4[j]);
                if(it != map.end()){ //找到
                    result += it->second;
                }
            }
        }
        return result;

    }
};

---

三、代码随想录学习

学习内容

思路完全一致。但是可能发现怎么参考给的代码简单呢？其实是使用成员函数不一样，获得的内容是同样的。

（1）使用map，可以用[ ] 下标运算符访问value。

map[key] ;
如果key不存在，可以创建一个；如果存在，map[key]++就是在操作value。

（2）使用range-for。遍历元素。C++11新加。如果有iterator，就可以用。

---

四、总结

重点在怎么引出unordered_map，容器里面放的是什么？

（欢迎指正，转载表明出处）