Leetcode P454 Java使用Hash表解决

Leetcode P454 Java使用Hash表解决

ideas

​ 由题意可知nums1[i] + nums2[j] + nums3[k] + nums4[l] == 0，我们可以分解为（nums1[i] + nums2[j]） + （nums3[k] + nums4[l]） == 0

我们使用HashMap来记录nums1[i] + nums2[j]相加的结果，每个结果个出现了多少次

        for (int i : nums1) {
            for (int j : nums2) {
                long tmp = i + j;
                map.put(tmp, map.getOrDefault(tmp, Long.valueOf(0))+1);
            }
        }

接下来判断0 - nums3[k] + nums4[l]是否在map中存在，如果存在则res+= map.get(0 - nums3[k] + nums4[l] )出现的个数，

关于为什么是0 - nums3[k] + nums4[l]？

​ 因为，假设nums1[i] + nums2[j]等于12，那么判断nums3[k] + nums4[l] 的值等于-12，0- （-12）查看12是否在map中存在，如果存在则res+= map.get(0 - nums3[k] + nums4[l] )出现的个数，

        for (int i : nums3) {
            for (int j : nums4) {
                long tmp = i+j;
                if (map.containsKey(0 - tmp)) {
                    res+=map.get(0-tmp);
                }
            }
        }

code

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int res =0;
        HashMap<Long,Long> map = new HashMap<>();
        for (int i : nums1) {
            for (int j : nums2) {
                long tmp = i + j;
                map.put(tmp, map.getOrDefault(tmp, Long.valueOf(0))+1);
            }
        }

        for (int i : nums3) {
            for (int j : nums4) {
                long tmp = i+j;
                if (map.containsKey(0 - tmp)) {
                    res+=map.get(0-tmp);
                }
            }
        }
        
        return res;
    }
}