给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请
你返回所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
输入:nums = [-1,0,1,2,-1,-4] 输出:[[-1,-1,2],[-1,0,1]] 解释: nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。 nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。 nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。 不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。 注意,输出的顺序和三元组的顺序并不重要。
解题思路:
//运用双指针,先将数组按升序排序,确定i的位置,j从i+1的位置开始往后走,k从n-1的位置往前坐
//sum>0,k--
//sum<0,j++
//sum=0,将nums[i],nums[j],nums[k]存入数组中
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
int n=nums.size();
sort(nums.begin(),nums.end());//对数组进行排序
int i=0;
for(int i=0;i<=n-3;i++)
{
if(i>0&&nums[i]==nums[i-1])//去掉重复值
continue;
int j=i+1;//j从i+1的位置开始
int k=n-1;//k从n-1的位置开始
while(j<k)//当j和k相遇时,结束循环
{
int sum=nums[i]+nums[j]+nums[k];//记录他们三个的和
if(sum>0)
k--;
else if(sum<0)
j++;
else
{
//当遇到数组中全是0的情况时,要注意输出的数组可能有重复
//eg.输入:[0,0,0,0] 输出:[0,0,0]
//所以要判断nums[j]是否等于nums[j-1],不然会输出两遍[0,0,0]
//j==i+1是除了全是0的情况,其余情况下,nums[j]==nums[j-1]时j==i+1也是符合条件
if(nums[j]!=nums[j-1]||j==i+1)
{
res.push_back({nums[i],nums[j],nums[k]});
}
j++;//j继续往后找,直到j和k相遇为止
}
}
}
//返回vector数组
return res;
}
};