力扣454 四数相加 2

给你四个整数数组 nums1、nums2、nums3 和 nums4 ，数组长度都是 n ，请你计算有多少个元组 (i, j, k, l) 能满足：

0 <= i, j, k, l < n
nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0

输入：4个数组。输出：满足条件的个数n

标签：数组 哈希

 /**
     * 思路：
     * 将前两个数组元素之和的种类和次数放在map的key value里
     * 将后两个数组元素之和的种类和次数放在map的key value里
     * 对一个map循环一下看数组里的key是否和另一个数组的key相反
     * @param nums1
     * @param nums2
     * @param nums3
     * @param nums4
     * @return
     */
    public static int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        HashMap<Integer, Integer> map = new HashMap<>();
        HashMap<Integer, Integer> map1 = new HashMap<>();
        int count = 0;
        int target ;
        for (int i = 0; i < nums1.length; i++) {
            for (int j = 0; j < nums2.length; j++) {
                target = nums1[i] + nums2[j];
                if(!map.containsKey(target)){
                    map.put(target,1);
                }
                else
                    map.put(target,map.get(target) + 1);
            }
        }

        for (int i = 0; i < nums3.length; i++) {
            for (int j = 0; j < nums4.length; j++) {
                target = nums3[i] + nums4[j];
                if(!map1.containsKey(target)){
                    map1.put(target,1);
                }
                else
                    map1.put(target,map1.get(target) + 1);
            }
        }
        for (Integer integer : map1.keySet()) {
            if(map.containsKey(-integer)){
                count += map.get(-integer) * map1.get(integer);
            }
        }

return count;
    }