@Aryami
题目
给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有和为 0 且不重复的三元组。
注意:答案中不可以包含重复的三元组。
示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
示例 2:
输入:nums = []
输出:[]
示例 3:
输入:nums = [0]
输出:[]
提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105
题解
【双指针法】一开始想用hashmap存储不同组合来解决,但是发现无法消除掉重复的三元组。所以,先将数组进行排序,排序后使用双指针法。先用 i i i固定第一个数字, l e f t left left表示 i i i后面一个数, r i g h t right right表示序列最后一个数。从两边向中间寻找符合条件的双元组。这里要注意, l e f t left left和 r i g h t right right都要进行查重处理。对于长度小于3的数组,可以特殊处理。时间复杂度为 O ( n 2 ) O(n^2) O(n2),空间复杂度为 O ( n ) O(n) O(n)(存储结果数组)。
class Solution {
public:
vector<vector<int>> threeSum(vector<int>& nums) {
vector<vector<int>> res;
if(nums.size()<3)
return {};
sort(nums.begin(), nums.end());
for(int i=0; i<nums.size(); i++)
{
// 双指针
int left = i + 1;
int right = nums.size() - 1;
if(i>0&&nums[i]==nums[i-1])
continue;
while(left<right)
{
if(left-1!=i && nums[left]==nums[left-1])
{
left++;
continue;
}
if(right+1<nums.size() && nums[right]==nums[right+1])
{
right--;
continue;
}
int sum = nums[left] + nums[right] + nums[i];
if(sum==0)
{
res.push_back({nums[left], nums[right], nums[i]});
left++;
right--;
}
else if(sum>0)
right--;
else
left++;
}
}
return res;
}
};