开启掘金成长之旅！这是我参与「掘金日新计划 · 2 月更文挑战」的第 20 天，点击查看活动详情

第454题.四数相加II

力扣题目链接(opens new window)

给定四个包含整数的数组列表 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 。所有整数的范围在 -2^28 到 2^28 - 1 之间，最终结果不会超过 2^31 - 1 。

例如:

输入:

A = [ 1, 2]
B = [-2,-1]
C = [-1, 2]
D = [ 0, 2]

输出:

解释:

两个元组如下:

(0, 0, 0, 1) -> A[0] + B[0] + C[0] + D[1] = 1 + (-2) + (-1) + 2 = 0
(1, 1, 0, 0) -> A[1] + B[1] + C[0] + D[0] = 2 + (-1) + (-1) + 0 = 0

思路

这道题乍一看也挺复杂的，是一道经典的哈希法题目。我觉得难点就在于要想到他是可以分组进行遍历的，把nums1和nums2分到一组，nums3和nums4分到一组，分别进行遍历，这样每个只需要用两层for循环，时间复杂度就是O(n^2)，这也是为什么我们不把1单分一组，234再分一组的原因，那样的话我们遍历后者就需要三层for循环，时间复杂度就变成了O(n^3)。

所以顺着上边的思路，就可以理出思路了，先遍历1和2，将他们的和都存入哈希表，统计记录他们出现的次数，然后遍历3和4的时候就只需要寻找哈希表中是不是存在0-t的数，如果有，就可以直接找到四数之和等于零的四个元素，直接用res加上他们出现的次数就可以了。

解法

java代码

class Solution {
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
        Map<Integer,Integer> map=new HashMap<>();
        int t;
        int res=0;
        for(int i:nums1){
            for(int j: nums2){
                t=i+j;
                if(map.containsKey(t)){
                    map.put(t,map.get(t)+1);
                }else{
                    map.put(t,1);
                }
            }
        }
        for(int i:nums3){
            for(int j:nums4){
                t=i+j;
                if(map.containsKey(0-t)){
                    res+=map.get(0-t);
                }
            }
        }
        return res;
    }
}

js代码

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @param {number[]} nums3
 * @param {number[]} nums4
 * @return {number}
 */
var fourSumCount = function(nums1, nums2, nums3, nums4) {
    const twoSumMap=new Map();
    let count=0;
    for(const i of nums1){
        for(const j of nums2){
            const t=i+j;
            twoSumMap.set(t,(twoSumMap.get(t) || 0)+1)
        }
    }
    for(const i of nums3){
            for(const j of nums4){
                 const t=i+j;
                 count+=(twoSumMap.get(0-t) || 0)
            }
    }
    return count;
};

刷题日记——四数之和||

第454题.四数相加II

思路

解法