Leetcode454 四数相加 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 。（来源力扣）
输入:
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

/*
	该题类似于采用两数相加问题，可以将A，B作为一组，C,D作为一组。
	将A，B两数组相加存入HashMap的键中，对应键重复出现的次数作为map的值。
	之后只需要判断C[i]+D[j]的相反数是否在map即可，该键对应的值*1就为出现
	0的次数，遍历完C和D，得到的sum累加值就是相应元组为0出现的总次数。
	
*/
public int fourSumCount(int[] A, int[] B, int[] C, int[] D) {
        int N=A.length;
        Map<Integer,Integer> map1=new HashMap<>();
        int x=1;
        for(int i=0;i<N;i++){
            for(int j=0;j<N;j++){
                if(map1.containsKey(A[i]+B[j])){
                    int num=map1.get(A[i]+B[j])+1;
                    map1.put(A[i]+B[j],num);
                }else{
                    map1.put((A[i]+B[j]),x);
                }
                
            }
        }    
        int sum=0;  
        for(int i=0;i<N;i++){
            for(int j=0;j<N;j++){
                if(map1.containsKey(-(C[i]+D[j]))){
                    sum+=map1.get(-(C[i]+D[j]));
                }             
            }
        } 
        return sum;
    }

类似题目：两数之和