Leetcode 454 - 4Sum II（hash）

题意

给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;
    }
};