問題描述
給定一個包含 n 個整數的數組 nums,判斷 nums 中是否存在三個元素 a,b,c ,使得 a + b + c = 0 ?找出所有滿足條件且不重複的三元組。
問題分析
基本思路就是先對數組排序,順序選擇一個數字,然後使用左右指針,依次對剩下數據進行遍歷,考慮到結果會有重複,所以,要注意如何去重,詳細思路請見源代碼。
源代碼
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums)
{
vector<vector<int>> res;
if(nums.size() <= 2) return res;
sort(nums.begin(), nums.end());
int l,r;
int sum=0;
for(int i=0;i<nums.size()-2;i++)
{
if(nums[i]>0) return res;
//如果前後兩個數相同的話,那麼最終找出的另外兩個數結果一定是相同的,所以可以直接跳過
if(i>0&&nums[i]==nums[i-1]) continue;
l=i+1;
r=nums.size()-1;
while(l<r)
{
sum=nums[i]+nums[l]+nums[r];
if(sum==0)
{
res.push_back({nums[i],nums[l],nums[r]});
l++;
r--;
//這裏也是爲了去重,在以i爲基礎,搜索後面兩個數的時候,面對重複的可以直接跳過,因爲最終得到的結果一定是相同的
while(l<r&&nums[l]==nums[l-1]) l++;
while(l<r&&nums[r]==nums[r+1]) r--;
}
else if(sum<0)
{
l++;
}
else
{
r--;
}
}
}
return res;
}
};