Leetcode#18 4Sum

原题地址

 

标准DFS+剪枝。

将数字提前排序，一来方便去重，二来创造剪枝条件。

1. 如果当前数字过大，就算加上最小的数字最后都超过target，停止继续尝试之后的数字，因为后面的数字更大

2. 如果当前数字过小，就算加上后面最大的数字最后都不到target，不用再考虑这个数字了，继续尝试后面更大的数字

3. 如果当前数字之前用过了，就直接跳过不再用了（去重）

 其他一些小优化：

寻找最后一个数字的时候可以二分查找

代码：

vector<vector<int> > res;
    
void dfs(vector<int> &num, vector<int> &ans, int pos, int left, int target) {
  if (left == 0) {
    if (target == 0)
      res.push_back(ans);
    return;
  }
        
  if (left == 1) {
    if (binary_search(num.begin() + pos, num.end(), target)) {
      ans.push_back(target);
      res.push_back(ans);
      ans.pop_back();
    }
    return;
  }
        
  set<int> record;
  for (int i = pos; i < num.size(); i++) {
    // 如果数字出现过，换后面的数字考虑
    if (record.find(num[i]) != record.end())
      continue;
    // 如果数字过大，肯定爆掉，跳出循环
    if (num[i] * left > target)
      break;
    // 如果数字太小，肯定够不着target，考虑后面更大的数字
    if (num[num.size() - 1] * (left - 1) < target - num[i])
      continue;
    record.insert(num[i]);
    ans.push_back(num[i]);
    dfs(num, ans, i + 1, left - 1, target - num[i]);
    ans.pop_back();
  }
}

vector<vector<int> > fourSum(vector<int> &num, int target) {
  vector<int> ans;
  sort(num.begin(), num.end());
  dfs(num, ans, 0, 4, target);
  return res;
}