【leetcode】454. 四数相加 II(medium)

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

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

思路：如果要暴力，那么时间复杂度将会是$O(n^4)$；可以通过两两数组分别遍历，那么就能将时间复杂度降到$O(n^2)$。具体的思路是：

• 先遍历nums1, nums2，将其所有可能的和都添加到HashMap中，同时记录重复出现的次数；
• 再遍历nums3, nums4，查找HashMap中是否存在(0-i-j)这个Key，该Key对应的Value就是能与(i,j)组成符合要求的四元组的情况数目。

解答：

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        HashMap<Integer,Integer> hm = new HashMap<>();
        int count = 0;
		// 将num1, num2数组两数之和及出现频率存储到HashMap
        for(int i: nums1)  
            for(int j: nums2)
                if(hm.containsKey(i+j))
                    hm.put(i+j, hm.get(i+j)+1);
                else
                    hm.put(i+j, 1);
		// 遍历计算num3,num4所有的和
        for(int i: nums3)
            for(int j: nums4)
                if(hm.containsKey(0-i-j))
                    count += hm.get(0-i-j);
        
        return count;
    }
}