LeetCode454 FourSumⅡ

LeetCode454 FourSumⅡ

题目描述：

给定四个包含整数的数组列表 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

解答：

解法一：双重循环+map查找表

基本思路是：一个hashmap来存放C,D所有和出现的次数，key为C,D的和，value出现的次数。两重循环遍历A,B，来确定一个和，去寻找map中0-sum出现的次数。

public static int solve(int[] A, int[] B, int[] C, int[] D) {
    int result=0;
    int targetSum =0, len = A.length;
    //将C,D和的每一种可能放入map中。
    Map<Integer, Integer> map = new HashMap<>();
    int sumCD;
    for (int i=0; i<len; i++){
        for (int j=0; j<len; j++){
            sumCD=C[i]+D[j];
            int count = map.getOrDefault(sumCD,0);
            map.put(sumCD,count+1);
        }
    }
    for (int i=0; i<len; i++) {
        for (int j = 0; j < len; j++) {
            int left = targetSum-A[i]-B[j];
            if (map.containsKey(left)){
                result+=map.get(left);
            }
        }
    }

    return result;
}