四数相加II Java

1.使用HashMap

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        //统计前两个数组中的元素之和(key)，以及出现次数(valuevalue)
        Map<Integer,Integer> map = new HashMap<>();
        int res = 0;
        for(int i : nums1){
            for(int j : nums2){
                int sum1 = i + j;
                if(map.containsKey(sum1)){
                    map.put(sum1,map.get(sum1)+1);
                }else{
                    map.put(sum1,1);
                }
            }
        }
        //统计剩余的两个元素的和，在map中找是否存在相加为0的情况，同时记录次数
        for(int m : nums3){
            for(int n : nums4){
                int sum2 = m + n;
                if(map.containsKey(0-sum2)){
                    res += map.get(0-sum2);
                }
            }
        }
        return res;
    }
}

优化

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        int res=0;       
        HashMap<Integer,Integer> map=new HashMap<>();
        for(int a:nums1){
            for(int b:nums2){
                map.put(a+b,map.getOrDefault(a+b,0)+1);
            }
        }
        // 先统计了前两个的和出现次数
        for(int c:nums3){
            for(int d:nums4){
                int temp=map.getOrDefault(-(c+d),-1);
                if(temp!=-1)
                    res+=temp;
            }
        }
        //再在后两个数组中元素的和的相反数中，在map里寻找，
        //是否包含其相反数，包含的话就说明有和为0的四个数，计数器计数
        //否则，循环进行下一次，寻找下一个和相反数
        //也可以先统计前三个数组各个元素的和的出现次数。
        return res;
    }
}