当前位置: 首页 > article >正文

C++ 优先算法 —— 三数之和(双指针)

目录

题目:三数之和 

1. 题目解析

2. 算法原理

①.  暴力枚举

②.  双指针算法

不漏的处理:

 去重处理:

 固定一个数 a 的优化:

3. 代码实现

Ⅰ.  暴力枚举(会超时 O(N³))

Ⅱ.  双指针算法 (O(N²))


题目:三数之和 

1. 题目解析

题目截图:

 

题目中所述的顺序不重要有两种情况:

  1. 最终选出来的几个三元组里,每个三元组里的数据顺序,是可以不用考虑的。
  2. 最终选出来的几个三元组,三元组的顺序,也是可以不用考虑的。

题目中还要求元素不能相同:

 

 

题目的细节要求也就是:

  • 最终选出来的几个三元组里面的元素是对应不同的 
  • 返回的三元组及其三元组里的数据的顺序不重要。 

  

题目给的示例二就是没有最终结果的情况

 

2. 算法原理

这道题有两种解法:

  • 暴力枚举(但会超时)
  • 双指针算法

①.  暴力枚举

虽然暴力枚举会超时,这里还要介绍一下的,因为双指针解法就是基于暴力枚举进行优化的。

这里方法就是:排序+暴力枚举(从左往右)+去重

 这里排序去重要注意的是:

所以可以换个策略,先对原数组整个排序:

所以,也就是先排序,再暴力枚举(从左往右枚举),再去重。

这样套三个for循环就可以暴力枚举结果了

//伪代码演示
for (int i = 0; i < n;) // 固定一个数a
{
    for (int j = i + 1; j < n;)     //依次固定枚举第二个数
    {
        for (int k = j + 1; k < n;)    //枚举第三个数
        {
            //查找符合的三元组,去重...
        }
    }
            
}

 (这里去重在下面双指针解法解释),这里套了三个for循环时间复杂度:O(N³)。

②.  双指针算法

这里双指针算法的解决方法就是:排序+双指针+去重

排序同上面暴力枚举,都先对一整个数组排序,再进行下面的处理。

这里举例一个已经排过序的新数组:

我们先固定一个数,这里先固定第一个数:

这里就转化成了和为s的两个数的问题,这里的s就是这个固定的数的相反数。也就是:

 所以这里解决步骤:

  1. 先对整个数组排序
  2. 固定一个数 a(这里有个小优化,后续介绍)
  3. 在该数后面的区间内,利用双指针算法快速找到两个数的和为 -a 的即可。

注意:这里只是找到了符合条件的情况,并没有去重。

接下来就是处理细节问题了,也就是去重确保所有符合条件的情况不落下(后面简称不漏) 。

和为s的两个数(这里会用到,详细介绍在此链接)

不漏的处理:

先控制指针找到符合情况:

这时可以看到是找到符合的情况了: 接下来就需要调整:

所以,也就是找到一种结果之后,不要停(两个指针不停),缩小区间(两指针都向内移动一步),继续寻找符合的情况。

 去重处理:

去重要考虑两种情况:

  1. 找到一种结果之后,left和right指针要跳过重复的元素(这里也就把第一个4给枚举完了)。但这里固定的 a 还是为-4,再继续枚举,是肯定都是重复结果的。
  2. 所以当使用完一次双指针算法后,a也需要跳过重复的元素。

这里去重可能指针会越界,为了避免越界:

这里的越界问题不止对整个数组区间的越界,还有对要用双指针处理的区间可能会越界。

在实现去重代码时需要加个判断条件就可以解决:

要判断处理这个条件,既可避免越界
if(left < right)        
也就判断两个指针是否在相遇前

 固定一个数 a 的优化:

因为我们前面先排序该数组为升序了,所以固定数a就可以仅仅需要保证它≤0即可。 

if (nums[i] > 0)
{
    break; // 小优化,数a为正数情况不用考虑,是一定不成立的
}

 接下来实现代码。

3. 代码实现

题目链接:三数之和

Ⅰ.  暴力枚举(会超时 O(N³))

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

        vector<vector<int>> ret; // 用来存储所有的三元组
        // 2.暴力枚举
        int n = nums.size();
        int i, j, k;
        for (i = 0; i < n;) // 固定一个数a
        {
            for (j = i + 1; j < n;)     //依次固定枚举第二个数
            {
                for (k = j + 1; k < n;)    //枚举第三个数
                {
                    int sum = nums[i] + nums[j] + nums[k];
                    if (sum == 0)
                    {
                        ret.push_back({ nums[i],nums[j],nums[k] });
                    }
                    
                    //去重
                    ++k;
                    while (k < n && nums[k] == nums[k - 1]) {
                        ++k;
                    }
                }
                //去重
                ++j;
                while (j < n && nums[j] == nums[j - 1]) {
                    ++j;
                }
            }
            //去重
            ++i;
            while (i < n && nums[i] == nums[i - 1]) {
                ++i;
            }
        }
        return ret;
    }
};

提交记录:

测试用例:

 

 

 

 

Ⅱ.  双指针算法 (O(N²))

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

        vector<vector<int>> ret; // 用来存储所有的三元组
        // 2.利用双指针解决
        int n = nums.size();
        for (int i = 0; i < n;) // 固定一个数a
        {
            if (nums[i] > 0) {
                break; // 小优化,数a为正数情况不用考虑,是一定不成立的
            }
            // 定义left,right,target通过双指针获取相加等于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里
                    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;
    }
};

注:这里时间复杂度是O(N²)。 

提交记录:

 

 制作不易,若有不足之处或出问题的地方,请各位大佬多多指教 ,感谢大家的阅读支持!!!  

 


http://www.kler.cn/a/385731.html

相关文章:

  • 遇到 mysql_config not found 错误
  • [ACTF2020]Upload 1--详细解析
  • 树莓派4B Qt+FFMPEG 多线程录制USB相机mjpeg数据流“h264_omx“硬件编码的MP4文件
  • 小程序19-微信小程序的样式和组件介绍
  • Vue3 模板语法
  • linux企业中常用NFS、ftp服务
  • 跨浏览器设备指纹
  • 拷贝构造(详解)
  • StarUML建模工具安装学习与汉化最新零基础详细教程【一键式下载】(适用于Windows、MacOS系统、Linux系统)
  • 哈尔滨等保测评常见误区破解:避免陷入安全盲区
  • 【人工智能】10分钟解读-深入浅出大语言模型(LLM)——从ChatGPT到未来AI的演进
  • 2024.6月GESP一级真题讲解(含视频讲解)
  • Spring学习笔记_28——事务
  • 【Qt】Macbook M1下载安装
  • 数据分析-43-时间序列预测之深度学习方法GRU
  • 【TS】九天学会TS语法——6.TypeScript泛型图文详解
  • 【Django】创建应用
  • Quartus Prime的应用
  • CentOS操作系统安装过程简介
  • Flutter 中 Provider 的使用指南
  • Python爬虫与Web渗透测试入门指南——初学者防踩雷
  • 现代Web开发:React Router 深度解析
  • MRCTF2020:千层套路
  • docker拉取和打包多个镜像
  • ros入门:参数服务器通信
  • pytorch torch.tile用法