Leetcode#16 3Sum Closest

原题地址

 

跟2Sum、3Sum、4Sum类似，都是：排序+搜索+剪枝

令sum = num[i] + num[j] + num[k] + (-target)（将-target看做一个必选的数），那这道题就跟4Sum（参见这篇文章）几乎一样了，变成了寻找最接近0的和。

需要剪枝的地方：

1. 数字太小，肯定不是最优解

2. 数字太大，肯定不是最优解

3. 数字重复

其他优化：

寻找最后一个数字可以用二分查找

 

代码：

int res;

void dfs(vector<int> &num, int ans, int pos, int left) {
  if (left == 0) {
    if (abs(ans) < abs(res))
      res = ans;
    return;
  }

  // 二分查找最后一个数
  if (left == 1) {
    int l = pos;
    int r = num.size() - 1;
    while (l <= r) {
      int m = (l + r) / 2;
      res = abs(ans + num[m]) < abs(res) ? ans + num[m] : res;
      if (ans + num[m] < 0)
        l = m + 1;
      else
        r = m - 1;
    }
    return;
  }

  unordered_set<int> old;
  for (int i = pos; i < num.size(); i++) {
    // 数字重复
    if (old.find(num[i]) != old.end())
      continue;
    // 数字太大
    if (num[i] * left + ans > abs(res))
      break;
    // 数字太小
    if (num[num.size() - 1] * (left - 1) + num[i] + ans < -abs(res))
      continue;
    old.insert(num[i]);
    dfs(num, ans + num[i], i + 1, left - 1);
  }
}

int threeSumClosest(vector<int> &num, int target) {
  sort(num.begin(), num.end());
  res = num[0] + num[1] + num[2] - target;
  dfs(num, -target, 0, 3);
  return res + target;
}