一、题目描述
Given an array nums
of n integers, are there elements a, b, c in nums
such that a + b + c = 0? Find all unique triplets in the array which gives the sum of zero.
Example:
Given array nums = [-1, 0, 1, 2, -1, -4], A solution set is: [ [-1, 0, 1], [-1, -1, 2] ]
二、思路
1.首先从小到大排序
2.然后遍历容器,i从0到nums.size(),并设两个指针,一个指向i+1,另一个指向nums.size()-1
3.遍历过程中,跳过重复的
4.注意输入为[0 0 0]的特殊情况
三、C++代码
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
std::sort(nums.begin(), nums.end());
for(int i = 0; i < nums.size(); i++) {
if(i > 0 && nums[i] == nums[i - 1]) continue; //相同则直接进行下次循环
int l = i + 1, r = nums.size() - 1;
while(l < r) {
int s = nums[i] + nums[l] + nums[r];
if(s > 0) r--;
else if(s < 0) l++;
else {
res.push_back(vector<int> {nums[i], nums[l], nums[r]});
while(l < r && nums[l] == nums[l + 1]) l++; //注意[0 0 0]特殊情况
while(l < r && nums[r] == nums[r - 1]) r--;
l++;
r--;
}
}
}
return res;
}
};