第454题.四数相加II

谁借莪１个温暖的怀抱￠ 2024-03-17 21:57 75阅读 0赞

#### 第454题.四数相加II ####

*  题干背景
 *  重要思路
 *  参考代码

## 题干背景 ##

给定四个包含整数的数组列表 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

## 重要思路 ##

1.  key放a和b两数之和，value放a和b两数之和出现的次数。
2.  时间复杂度: O(n^2)
3.  空间复杂度: O(n^2)

## 参考代码 ##

// 454. 四数相加 II
    public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4){
        
        int res = 0;
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        for(int i : nums1){
        
            // 统计两个数组中的元素之和，同时统计出现的次数，放入map
            for(int j : nums2){
        
                int sum = i + j;
                map.put(sum, map.getOrDefault(sum, 0)+1);
            }
        }
        // 统计剩余的两个元素的和，在map中找是否存在相加为0的情况，同时记录次数
        for(int i : nums3){
        
            for(int j : nums4){
        
                res += map.getOrDefault(0-i-j, 0);
            }
        }
        return res;
    }

**参考资料**：[第454题.四数相加II][454_._II]

[454_._II]: https://programmercarl.com/0454.%E5%9B%9B%E6%95%B0%E7%9B%B8%E5%8A%A0II.html#%E5%85%B6%E4%BB%96%E8%AF%AD%E8%A8%80%E7%89%88%E6%9C%AC