第一次写用了dfs,做了一定优化,但是还是TLE
原来这种题有套路解法,用两个游标相对寻找会节省很多时间
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
set<vector<int>> res;//set不允许存储相同的元素,达到去重的效果
sort(nums.begin(), nums.end());//压缩时间复杂度第一处
for (int k = 0; k < nums.size(); ++k) {
if (nums[k] > 0) break;//处理corner case,同时也是压缩时间复杂度的第二处
int target = 0 - nums[k];
int i = k + 1, j = nums.size() - 1;
while (i < j) {
if (nums[i] + nums[j] == target) {
res.insert({ nums[k], nums[i], nums[j] });
while (i < j && nums[i] == nums[i + 1]) ++i;//去重,压缩时间复杂度第三处
while (i < j && nums[j] == nums[j - 1]) --j;
++i; --j;
}
else if (nums[i] + nums[j] < target) ++i;
else --j;
}
}
return vector<vector<int>>(res.begin(), res.end());
}
};