【算法笔记】四数相加

四数相加问题

给定四个包含整数的数组列表 A , B , C , D ,计算有多少个元组 (i, j, k, l) ，使得 A[i] + B[j] + C[k] + D[l] = 0。

为了使问题简单化，所有的 A, B, C, D 具有相同的长度 N，且 0 ≤ N ≤ 500 。所有整数的范围在 -228 到 228 - 1 之间，最终结果不会超过 231 - 1 。

问题分析

看过这道题目后，第一反应就是找到四个数组所有的组合方式，判断相加是否为零，是则保存结果，这种做法的时间复杂度非常大，显然是不可取的方式。

注意题目中只让我们找到有多少个满足条件的组合方式，并没有让我们记录组合的方式，所以只需要关注有多少个组合就是。

如果我们将四个数组一起计算，那么时间复杂度还是很大，所以我们分组计算，两两一组，分别计算和的大小和每一种和情况的次数，然后再作比较。

上述说到需要记录和的大小和该大小的出现次数，这两个数为对应关系，可采用map集合来存储。
基本操作过程为

• 使用map存储num1和mun2数组中所有组合的大小与对应大小出现的次数
• 判断map中num1与num2和的大小减去num3与num4和的大小是否为零，这里可以采用判断num3与num4和的负数是否存在于map中。

代码

public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
    //创建哈希表
    Map<Integer, Integer> hashTable=new HashMap();
    //记录满足的条件的结果数目
    int count=0;

    //遍历nums1
    for(int i:nums1){
        //遍历nums2
        for(int j:nums2){
            //将nums1与nums2的元素所有组合的和放入哈希表中，并放入和出现的次数，使用hashTable.getOrDefault获取已保存的大小然后加1
            hashTable.put(i+j,hashTable.getOrDefault(i+j,0)+1);
        }
    }


    //遍历nums3
    for(int i:nums3){
        //遍历nums4
        for(int j:nums4){
            //判断是否有满足条件的组合
            if(hashTable.containsKey(-i-j)){
                //存储结果次数
                count+=hashTable.get(-i-j);
            }
        }
    }

    return count;
}