static const auto ______ = [](){
std::cout.sync_with_stdio(false);
cin.tie(0);
return 0;
}();
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
if (nums.size() < 3)
return vector<vector<int>>();
// 先排序
sort(nums.begin(), nums.end());
vector<vector<int>> ans;
int a, b, c;
int size = nums.size();
for (a = 0; a < size - 2; ++a){
// 如果a大于0了,直接返回
if (nums[a] > 0)
return ans;
b = a + 1;
c = size - 1;
while(b < c){
// 如果符合要求
if (nums[a] + nums[b] + nums[c] == 0){
ans.push_back({nums[a], nums[b], nums[c]});
b++,c--;
// 跳过和b相同的数字
while (b < c && nums[b] == nums[b - 1]) b++;
// 跳过和c相同的数字
while (b < c && nums[c] == nums[c + 1]) c--;
}else if (nums[a] + nums[b] + nums[c] < 0 )
b++;
else
c--;
}
// 跳过和a相同的数字
while(a < size && nums[a + 1] == nums[a]) a++;
}
return ans;
}
};
LetCode 15. 三数之和
最新推荐文章于 2024-08-27 14:15:34 发布