哈希表系列⑤ -- 四数相加 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
示例：
输入：nums1 = [1,2], nums2 = [-2,-1], nums3 = [-1,2], nums4 = [0,2]
输出：2
解释：
两个元组如下：
(0, 0, 0, 1) -> nums1[0] + nums2[0] + nums3[0] + nums4[1] = 1 + (-2) + (-1) + 2 = 0
(1, 1, 0, 0) -> nums1[1] + nums2[1] + nums3[0] + nums4[0] = 2 + (-1) + (-1) + 0 = 0
示例：
输入：nums1 = [0], nums2 = [0], nums3 = [0], nums4 = [0]
输出：1

解题思路

在不考虑暴力解法的同时，我们可以有这样的思路：将前两个数组所有加和的结果进行记录存储(这个过程用两个for循环嵌套即可)，然后再将后两组的加和结果和前两组的进行对比(每计算一次就对比一次)，看是否有相反数，将互为相反数的进行计数，最后返回结果即可。

这其中也存在这一个问题：用什么数据结构去存储前两个数组的所有加和结果？

• 用Set显然不行，如果有两种情况的和是一样的，只会显示一个，后面进行计数的时候会产生误差。
• 用列表可以，不过效率不高。因为到最后去计数的时候，你要一个个地去遍历符合条件的相反数有几个。直接将时间复杂度上升了一个层级。
• 用map是最好的。我们可以将和的结果作为key，而将出现次数作为关联的值，如此我们最后计数的时候只需要进行简单的累加就可以了，完全不需要再去进行遍历来计数。

代码实现

public int fourSumCount(int[] a, int[] b, int[] c, int[] d) {
		Map<Integer,Integer> box = new HashMap<>();
		int result = 0;
		for (int num1 : a) {
			for (int num2 : b) {
				box.put(num1 + num2, box.getOrDefault(num1 + num2, 0) + 1);
			}
		}  
		for (int num3 : c) {
			for (int num4 : d) {
				if (box.containsKey(-(num3 + num4))) {
					result += box.get(-(num3 + num4));
				}
			}
		}   
		return result;

看起来有点臃肿，我们可以优化以下代码：

class Solution {
    public int fourSumCount(int[] a, int[] b, int[] c, int[] d) {
        Map<Integer,Integer> box = new HashMap<>();
		int result = 0;
		for (int num1 : a) for (int num2 : b) box.put(num1 + num2, box.getOrDefault(num1 + num2, 0) + 1);
		for (int num3 : c) for (int num4 : d) if (box.containsKey(-(num3 + num4))) result += box.get(-(num3 + num4));
		return result;
    }
}