leetcode_middle_18_454. 4Sum II

最新推荐文章于 2025-02-03 15:37:10 发布

pusude

最新推荐文章于 2025-02-03 15:37:10 发布

阅读量240

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/pusude/article/details/54879960

leetcode 专栏收录该内容

183 篇文章

订阅专栏

本文介绍了一种解决特定问题的方法：给定四个相同长度的整数数组，找到所有可能的四元组组合，使得这些组合的元素之和等于0，并计算这样的组合总数。通过使用哈希映射减少搜索范围，将原本的四重循环转化为更高效的两重循环。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：

给四个长度相等的整数数组，每个数组取一个数相加，等于0.这样的组合有多少个？

分析：

我们要全部搜索遍历的话需要四重循环，有没有什么办法不四重循环呢，首先想到的是重复出现的可以用map来存放它出现的次数，但是这个并不能减少循环次数。继续思考，没什么头绪，我们来观察题目特征。我们要的是和为a+b+c+d等于0意味着四重循环 ,可以推出：a+b+c = -d , a+b = -c - d...是不是意味着三重循环加一重循环，二重循环加二重循环呢？我们继续整理思路，选取最好的两个二重循环：

public class Solution {
    public int fourSumCount(int[] A, int[] B, int[] C, int[] D) {
        Map<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;
    }
}