【算法专题】双指针—三数之和

力扣题目链接:三数之和

一、题目解析

二、算法原理

解法一:排序+暴力枚举+利用set去重

代码就不写了,你们可以试着写一下

解法二:排序+双指针

这题和上一篇文章的两数字和方法类似

  1. 排序
  2. 固定一个数a
  3. 在这个数的后面区间,使用双指针找到两个数之和为-a即可

需要解决两个细节问题:

1. 去重(避免重复的三元组)

  • 找到一种结果后left和right要跳过重复的元素
  • 当双指针使用完跳出循环后,a也需要跳过重复的元素

去重的时候还要控制边界,避免越界的问题

2. 不漏(不漏掉任何一个满足条件的三元组)

找到一种结果后不要"停"(left++,right--),缩小区间,继续往后寻找。

三、代码编写

cpp 复制代码
class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> ret;
        //1.排序
        sort(nums.begin(), nums.end());

        int n = nums.size();
        //2.双指针
        for(int i = 0; i < n; )
        {
            if(nums[i] > 0) break;//小优化:大于0的直接退出循环不用往后找了
            int left = i + 1, right = n - 1, target = -nums[i];
            while(left < right)
            {
                int sum = nums[left] + nums[right];
                if(sum > target) right--;
                else if (sum < target) left++;
                else
                {
                    ret.push_back({nums[i], nums[left], nums[right]});
                    left++;
                    right--;
                    //去重left和right
                    while(left < right && nums[left] == nums[left - 1]) left++;
                    while(left < right && nums[right] == nums[right + 1]) right--; 
                }
            }
            i++;
            //去重i
            while(i < n && nums[i] == nums[i - 1]) i++;
        }
        return ret;
    }
};

这里还有个小优化,当a大于0的话可以不用在往后找,因为后面都是大于0的数,再怎么找也不会满足条件的。

相关推荐
倒头就睡的小比特1 天前
算法竞赛C++常用的STL
c++·算法
weilx12341 天前
C++笔记-文件IO-<fcntl.h>
c++
小羊没烦恼!1 天前
初探性能优化——2个月到4小时的性能提升
java·开发语言·windows·算法·c#
猎头南楼1 天前
知识社区推荐系统实践:新用户冷启动与长短期兴趣建模的挑战 资深推荐算法工程师
人工智能·深度学习·算法·机器学习
Smileyqp沛沛1 天前
前端?C++ ?较大差异基础罗列
c++·基础·前端转c++
C语言小火车1 天前
C/C++ 为什么需要编译器?
开发语言·c++
旖旎夜光1 天前
力控面试题 01.01: 判定字符是否唯一(位运算) —— 题解
c++·学习·算法·leetcode·力控
wzdark1 天前
大规模并行计算中的负载均衡算法研究4
算法
吞下星星的少年·-·1 天前
C++ 萌新语法入门篇
c++·算法比赛
Because_of_Her11 天前
并查集-听课笔记
笔记·算法·并查集