LeetCode---454. 4Sum II

LeetCode—454. 4Sum II

题目

https://leetcode.com/problems/4sum-ii/
给出四个数组，长度相同，从每个数组中挑一个数作和，判断是这个和为零的挑选方式一共有几种。

思路及解法

这道题可以和数组的前面几道题一块看，包括TwoSum 和 4Sum。
4Sum是在说一个数组中找出四个数使的和为target，那么这道题的区别就是变成了四个数组，那肯定不能直接把四核数组合成一个数组做，毕竟是一个数组取一个数。
最简单暴力的解法就是四个循环了，但复杂度太高了，不可取。但我们可以沿着这个思路往下想，如果能把四个循环拆开，复杂度就降低了。怎么拆呢？TwoSum中用到了hashmap，就是答案。将四个循环拆成两个循环。
本质是扎样的，我们想四个循环的额情况，看成两部分的话，我们是拿后面的数去匹配前面得到的和，如果是相反数就是合理的解。这样看来，实际上四层的循环嵌套是可以分开的，只要用后面的匹配前面就行了。那么怎么拆分复杂度最低呢？当然就是拆成两个二层循环了。拆分途径依赖的就是hashmap
先前两个数组元素的和放到map中，key是两数之和，value是和出现的次数。后两个数组的和我们去看他是否与map中的key相反数，如果是，就是一个合理的解，我们计数。

代码

class Solution {
    public int fourSumCount(int[] A, int[] B, int[] C, int[] D) {
        HashMap<Integer, Integer> map = new HashMap<>();
        int count = 0;
        for(int i=0; i<A.length; i++){
            for(int j=0; j<B.length; j++){
                int sum1 = A[i]+B[j];
                if(map.containsKey(sum1))
                    map.put(sum1, map.get(sum1)+1);
                else
                    map.put(sum1, 1);
            }
        }
        for(int i=0; i<C.length; i++){
            for(int j=0; j<D.length; j++){
                int sum2 = C[i]+D[j];
                if(map.containsKey(-sum2))
                    count += map.get(-sum2);
            }
        }
        return count;
    }
}