Leetcode 454 - 4Sum II（hash）

最新推荐文章于 2023-03-19 12:54:46 发布

Lzed

最新推荐文章于 2023-03-19 12:54:46 发布

阅读量245

点赞数

CC 4.0 BY-SA版权

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

本文链接：https://blog.youkuaiyun.com/Lzedo/article/details/54976293

leetcode 同时被 3 个专栏收录

157 篇文章

订阅专栏

二分

27 篇文章

订阅专栏

hash

8 篇文章

订阅专栏

本文介绍了一种高效解决四数求和问题的方法，通过预处理将两两数组的组合结果存储，利用哈希表实现快速查找匹配，最终达到O(n²)的时间复杂度。

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

题意

给4个长度相同的数组，在每个数组里面选一个数，使它们的和加起来为0。问一共有多少方案？

思路

最先想的是 $O(n^3logn)$ 的。就ABC枚举一下，D二分一下。

但是，我们可以考虑：先将AB的所有和 $n^2$ 枚举一下放到一个新数组a里面，CD的和 $n^2$ 加入到unordered_map里面，然后遍历a里面的元素，去unordered_map里面 $O(1)$ 查找一下就好了。

时间复杂度： $O(n^2)$

代码

class Solution {
public:
    int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
        int n = A.size();
        vector<int> a;
        unordered_map<int, int> b;
        for (int i = 0; i < n; i++) {
            for (int j = 0; j < n; j++) {
                a.push_back(A[i] + B[j]);
                b[C[i] + D[j]]++;
            }
        }
        int tot = 0;
        for (auto x : a) {
            if (b.find(-x) != b.end()) tot += b[-x];
        }
        return tot;
    }
};