class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
if (nums.size() < 3) return {};
vector<vector<int>> res;
sort(nums.begin(), nums.end());
for (int i = 0; i < nums.size() - 2; i++) {
if (i > 0 && nums[i] == nums[i - 1]) continue;
// 在 i + 1 ... nums.length - 1 中查找相加等于 -nums[i] 的两个数
int target = - nums[i];
int left = i + 1, right = nums.size() - 1;
while (left < right) {
int sum = nums[left] + nums[right];
if (sum == target) {
res.push_back({nums[i], nums[left], nums[right]});
/*
下面的代码相当于:
while (left < right) {
// 不管前后相不相等,left 都要往前走
left++;
if (nums[left - 1] != nums[left]) break;
}
while (left < right) {
// 不管前后相不相等,right 都要往后走
right--;
if (nums[right + 1] != nums[right]) break;
}
*/
// 去重
while (left < right && nums[left] == nums[++left]);
while (left < right && nums[right] == nums[--right]);
} else if (sum < target) {
left++;
} else {
right--;
}
}
}
return res;
}
};
作者:tangweiqun
链接:https://leetcode-cn.com/problems/1fGaJU/solution/jian-dan-yi-dong-javac-pythonjs-san-shu-nu6el/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
本菜鸡写的没有去重功能的代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> ret;
if(nums.size() < 3)
return ret;
sort(nums.begin(),nums.end());
for(int i = 0;i < nums.size()-2;i++){
int left = i + 1;
int right = nums.size()-1;
while(left < right){
//int mid = left + (right - left)/2;
int sum = nums[i] + nums[left] + nums[right];
if(sum == 0){
ret.push_back({nums[i],nums[left],nums[right]});
// ret.push_back();
// ret.push_back();
//找到一组之后应该如何处理呢?
//又该如何去重呢?
left++;
}
else if(sum > 0){
right--;
}
else if(sum < 0){
left++;
}
}
}
return ret;
}
};
关于去重有3个部分:
1)每次寻找一个起点的时候可能数字会重复