class Solution {
public:
int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) {
unordered_map<int,int> hashmap;
int ans = 0;
for(int i=0;i<A.size();i++){
for(int j=0;j<B.size();j++){
if(hashmap.count(A[i]+B[j])<1)
hashmap[A[i]+B[j]] = 1;
else
hashmap[A[i]+B[j]] += 1;
}
}
for(int i=0;i<C.size();i++){
for(int j=0;j<D.size();j++){
if(hashmap.count(-(C[i]+D[j]))>0)
ans += hashmap[-(C[i]+D[j])];
}
}
return ans;
}
};
这是leetcode四数之和II的代码,思想非常非常简单,只是二数之和的拓展,用空间换取时间的思路。
三数之和可以用同样的做法。
思想继续升级,如果是六数之和呢?
如果是八数之和呢?
你肯定会说是O(n^4)循环一波了吧?不过答案是否定的,这样浪费时间,我们只要2次方的时间复杂就ok了。思路如下:
先2次方计算出八个数组(假定编号1,2,3,4,5,6,7,8)的1+2,3+4,5+6,7+8共4组组合,此时用掉了O(n^2)的时间,
再计算(1+2) + (3+4) , (5+6) + (7+8) 的组合结果,同样用掉了O(n^2) 的时间,依次类推。。。。
你会发现,只要是N数之和(注意,是给定多个数组的那种,而不是给定一个数组),我们都可以在O(N^2)的时间内解决掉,不断归并就是了。
N数之和II的变种可以是一个数组,但必须要求可以取重复。