LeetCode454—四数相加Ⅱ（java版）

最新推荐文章于 2021-10-20 17:18:32 发布

陈粑粑的小白鞋

最新推荐文章于 2021-10-20 17:18:32 发布

阅读量122

点赞数

分类专栏： LeetCode 文章标签：哈希表二分查找整数数组元组四数之和

本文链接：https://blog.csdn.net/qq_40840749/article/details/115253586

版权

LeetCode 专栏收录该内容

187 篇文章 2 订阅

订阅专栏

题目描述：

标签：哈希表二分查找

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

代码：

思路分析：

1、定义一个map，key存放A[I]+B[j]的和，value存放和出现的次数，其中利用了map.getOrDefault(key,default)方法，如果key存在则返回对应的value值，不存在则返回默认值default。

2、遍历A和B数组，将和、出现的次数加入map中

3、遍历C和D数组，看0-C[k]-D[l]是否存在map中，如果存在count += 次数。

class Solution {
    public int fourSumCount(int[] A, int[] B, int[] C, int[] D) {
        Map<Integer,Integer> map = new HashMap<>();
        for(int n1 : A){
            for( int n2 : B){
                map.put(n1 + n2,map.getOrDefault(n1 + n2,0) + 1);
            }
        }
        int count = 0;
        for(int n3 : C){
            for(int n4 : D){
                if(map.containsKey(0 - n3 -n4)){
                    count += map.get(0 - n3 - n4);
                }
            }
        }

        return count;
    }
}