15. 三数之和
https://leetcode.cn/problems/3sum/
这道题就是很典型的双指针的思想,关键地方就在于如何去重。
1.题目保证数组nums最少有三个元素,所以不用考虑元素个数小于三的情况。
2.排序,使得数组从小到大排序,若nums[0]>0则可以直接返回结果。
3.令左指针 left=k+1,右指针 right=n−1(k是从0开始向后遍历)
若nums[k]+nums[left]+nums[right]==0,将其加入结果集。然后进行去重操作,也就是把left向右移动到不等于nums[left-1]的位置,同样right要向前移动到不等于nums[right+1]的位置。然后left向后遍历,right向前遍历。
若和大于 0,说明 nums[right]过大,right指针左移。
若和小于 0,说明 nums[left] 太小,left指针 右移。
最后对指针k去重,然后向后遍历
具体代码实现
vector<vector<int>> threeSum(vector<int>& nums)
{
sort(nums.begin(),nums.end());
vector<vector<int>>ans;
int k=0;
while(k<nums.size())
{
if(nums[k]>0)break;
int i=k+1,j=nums.size()-1;
while(i<j)
{
if(nums[i]+nums[j]+nums[k]>0)j--;
else if(nums[i]+nums[j]+nums[k]<0)i++;
else
{
ans.push_back({nums[i],nums[k],nums[j]});
while(i<j&&nums[i]==nums[i+1])i++;
while(i<j&&nums[j]==nums[j-1])j--;
i++;
j--;
}
}
while(k+1<nums.size()&&nums[k]==nums[k+1])k++;
k++;
}
return ans;
}