思路:排序+双指针
1:去重:先递增排序,碰到跟之前相同的继续移动。
2:如果和最小的两个元素相加 和 大于0 直接break:
3:如果和最大的两个元素相加 和 小于0 直接continue进入下一轮循环
注意:如果元素>0则break没什么意义,因为3一定会满足2。。。。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> ans;
int n = nums.size();
sort(nums.begin(), nums.end());
for (int i = 0; i < n - 2; ++i) {
if (i - 1 >= 0 && nums[i] == nums[i - 1]) continue; //去重:1
if (nums[i] + nums[i + 1] + nums[i + 2] > 0) break; //加两个最小的大于0 直接break---------------剪枝
if (nums[i] + nums[n - 1] + nums[n - 2] < 0) continue;
//加两个最大的小于0 直接continue---------------剪枝
int j = i + 1, k = n - 1;
while (j < k) {
int sum = nums[i] + nums[j] + nums[k];
if (sum < 0) {
++j;
while (j < k && nums[j] == nums[j - 1]) ++j; //加上条件:j < k
}else if (sum > 0){
--k;
while (j < k && nums[k] == nums[k + 1]) --k;
}else {
ans.push_back({nums[i], nums[j], nums[k]});
++j;
while (j < k && nums[j] == nums[j - 1]) ++j; //==0的时候要处理去重:2
--k;
while (j < k && nums[k] == nums[k + 1]) --k;
}
}
}
return ans;
}
};