模块一——双指针:15.三数之和

文章目录

  • 题目描述
  • 算法原理(排序+双指针)
  • 代码实现

题目描述

题目链接:15.三数之和
模块一——双指针:15.三数之和_第1张图片

算法原理(排序+双指针)

本题与两数之和类似,是⾮常经典的⾯试题。与两数之和稍微不同的是,题⽬中要求找到所有不重复的三元组。那我们可以利⽤在两数之和那⾥⽤的双指针思想,来对我们的暴⼒枚举做优化:

  1. 先排序;
  2. 然后固定⼀个数a
  3. 在这个数后⾯的区间内,使⽤双指针算法快速找到两个数之和等于-a 即可。

但是要注意的是,这道题⾥⾯需要有去重操作~

  • 找到⼀个结果之后, left 和right 指针要跳过重复的元素;
  • 当使⽤完⼀次双指针算法之后,固定的a 也要跳过重复的元素。

代码实现

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> vv;
        sort(nums.begin(),nums.end());//1.排序
        for(int i = 0;i < nums.size() - 1;)//2.固定一个数
        {
            if(nums[i] > 0)break;//小优化
            int left = i + 1,right = nums.size() - 1,target = -nums[i];
            while(left < right)//3.利用双指针算法快速找出等于|nums[i]|的组合
            {
                int sum = nums[left] + nums[right];
                if(sum > target)--right;
                else if(sum <target)++left;
                else
                {
                    vv.push_back({nums[i],nums[left],nums[right]});
                    ++left,--right;
                    while(left < right&& nums[left] == nums[left - 1])++left;//去重left
                    while(left < right&& nums[right] == nums[right + 1])--right;//去重right    
                }
            }
            ++i;
            while(nums[i] == nums[i - 1]&&i < nums.size() - 1)++i;//去重
        }
        return vv;
    }
};

你可能感兴趣的:(算法学习之路,算法,c语言,c++)