leetcode-454：四数相加Ⅱ

问题

给定四个包含整数的数组列表 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 。所有整数的范围在到之间，最终结果不会超过。

输入：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

思路

• 首先定义 一个map，key放a和b两数之和，value放a和b两数之和出现的次数
• 遍历A和B数组，统计两个数组元素之和，和出现的次数，放到map中
• 定义变量ans，用来统计a+b+c+d = 0 出现的次数。
• 遍历C和D数组，找到如果 0-(c+d) 在map中出现过的话，就用ans把map中key对应的value也就是出现次数统计出来
• 最后返回统计值 count 就可以了 ```java package leetcode;

import java.util.HashMap; import java.util.Map;

public class leetcode_454 { public int fourSumCount(int[] A, int[] B, int[] C, int[] D) { Map countAB = new HashMap(); // 遍历A和B数组，统计两个数组元素之和，和出现的次数，放到map中 for (int u : A) { for (int v : B) { /getOrDefault()方法获取指定key对应对value，如果找不到key，则返回设置的默认值 put()方法将指定的键/值对插入到HashMap中，如果插入的 key 对应的 value 已经存在， /则执行 value 替换操作，返回旧的 value 值 countAB.put(u + v, countAB.getOrDefault(u + v, 0) + 1); } } int ans = 0; // 遍历C和D数组，找到如果 0-(c+d) 在map中出现过的话，就把map中key对应的value也就是出现次数统计 出来 for (int u : C) { for (int v : D) { if (countAB.containsKey(-u - v)) { ans += countAB.get(-u - v); } } } return ans; } } ```