题意简述:给出A、B、C、D四个区间,分别从这四个区间取出一个整数,使其相加得到的和等于0,求出总共有多少种方式。 输入:四个整型的vectorA、B、C、D 输出:满足条件的四元组(i,j,k,l)的个数
题解: 如果采用四个区间同时进行枚举的方式,时间复杂度是O(N 4 ),这显然是不可接受的。注意到枚举的对象既可以是区间内的整数,也可以是整数的和。所以可以采用二分的思路:把四个区间分成两组:A、B一组,C、D一组,先从A、B组枚举出从两组中各取一个整数相加所有可能的结果,将这些和排序(将这个排好序的vector记为ab),再枚举出C、D一组的所有情况,每算出一个c+d,就在ab中寻找是否存在c+d的相反数,由于ab是已排序的,所以可以采用二分搜索的方式。考虑到ab可能存在重复的和,所以利用标准库的upper_bound 和lower_bound来求出重复的数有多少个。
空间复杂度:主要贡献是vector ab,O(N 2 ) 时间复杂度:ab求和的复杂度O(N 2 ),C、D枚举加上二分搜索的复杂度O(N 2 logN),所以总的时间复杂度是O(N 2 logN)
int fourSumCount(vector<int>& A, vector<int>& B, vector<int>& C, vector<int>& D) { vector<int> ab; int result = 0; for(int i = 0;i < A.size();i++) { for(int j = 0;j < B.size();j++) { ab.push_back(A[i]+B[j]); } } sort(ab.begin(), ab.end()); for(int i = 0;i < C.size();i++) { for(int j = 0;j < D.size();j++) { int temp = -(C[i]+D[j]); result += upper_bound(ab.begin(),ab.end(),temp) - lower_bound(ab.begin(),ab.end(),temp); } } return result; }