标签：
标准DFS+剪枝。
将数字提前排序，一来方便去重，二来创造剪枝条件。
1. 如果当前数字过大，就算加上最小的数字最后都超过target，停止继续尝试之后的数字，因为后面的数字更大
2. 如果当前数字过小，就算加上后面最大的数字最后都不到target，不用再考虑这个数字了，继续尝试后面更大的数字
3. 如果当前数字之前用过了，就直接跳过不再用了（去重）
其他一些小优化：
寻找最后一个数字的时候可以二分查找
代码：
1 vector<vector<int> > res; 2 3 void dfs(vector<int> &num, vector<int> &ans, int pos, int left, int target) { 4 if (left == 0) { 5 if (target == 0) 6 res.push_back(ans); 7 return; 8 } 9 10 if (left == 1) { 11 if (binary_search(num.begin() + pos, num.end(), target)) { 12 ans.push_back(target); 13 res.push_back(ans); 14 ans.pop_back(); 15 } 16 return; 17 } 18 19 set<int> record; 20 for (int i = pos; i < num.size(); i++) { 21 // 如果数字出现过，换后面的数字考虑 22 if (record.find(num[i]) != record.end()) 23 continue; 24 // 如果数字过大，肯定爆掉，跳出循环 25 if (num[i] * left > target) 26 break; 27 // 如果数字太小，肯定够不着target，考虑后面更大的数字 28 if (num[num.size() - 1] * (left - 1) < target - num[i]) 29 continue; 30 record.insert(num[i]); 31 ans.push_back(num[i]); 32 dfs(num, ans, i + 1, left - 1, target - num[i]); 33 ans.pop_back(); 34 } 35 } 36 37 vector<vector<int> > fourSum(vector<int> &num, int target) { 38 vector<int> ans; 39 sort(num.begin(), num.end()); 40 dfs(num, ans, 0, 4, target); 41 return res; 42 }
标签：
原文地址：http://www.cnblogs.com/boring09/p/4262839.html
