题意
给4个长度相同的数组,在每个数组里面选一个数,使它们的和加起来为0。问一共有多少方案?
思路
最先想的是 O(n3logn) 的。就ABC枚举一下,D二分一下。
但是,我们可以考虑:先将AB的所有和
n2
枚举一下放到一个新数组a里面,CD的和
n2
加入到unordered_map
里面,然后遍历a里面的元素,去unordered_map
里面
O(1)
查找一下就好了。
时间复杂度: O(n2)
代码
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;
}
};