LeetCode 454 4Sum II 题解

题意简述：给出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;
}