LeetCode 454 四数相加II

Leetcode 454 四数相加II

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

执行用时：73 ms, 在所有 Java 提交中击败了89.61%的用户
内存消耗：59.3 MB, 在所有 Java 提交中击败了53.65%的用户

拆分
将四数和(a,b,c,d)拆分为两两之和(a,b; c,d)，则a+b+c+d=0可转化为a+b = -(c+d)。先寻找所有可能的a, b组合并对其和进行统计，在对每一个c, d组合寻找到和为-(c+d)的a, b组合的个数加入到最后的结果count中。

class Solution {
    public int fourSumCount(int[] A, int[] B, int[] C, int[] D) {
        //边界情况: 存在数组为空
        if( (A==null || A.length==0) 
            ||(B==null || B.length==0)
            ||(C==null || C.length==0)
            ||(D==null || D.length==0)) {
            return 0;
        }

        //遍历A,B数组统计所有的组合的和、和出现的次数
        HashMap<Integer, Integer> map = new HashMap<>();
        for(int i=0; i<A.length; i++) {
            for(int j=0; j<B.length; j++) {
                map.put(A[i]+B[j], map.getOrDefault(A[i]+B[j], 0)+1);
            }
        }

        //遍历C,D数组找到 -(c+d) = a+b的情况
        int count = 0;
        for(int i=0; i<C.length; i++) {
            for(int j=0; j<D.length; j++) {
                count += map.getOrDefault(-1*(C[i]+D[j]), 0);
            }
        }

        return count;
    }
}