leetcode三数之和C++版

最新推荐文章于 2024-06-02 00:17:07 发布

原创最新推荐文章于 2024-06-02 00:17:07 发布 · 868 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode #三数之和 #C++

c++ 专栏收录该内容

50 篇文章

订阅专栏

该博客讨论了LeetCode中的三数之和问题，提供了使用C++的两种解题方法：暴力法（时间复杂度较高，O(n^3)）和指针法。指针法通过固定一个数字，然后从数组两端遍历寻找满足条件的三元组，以降低时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。

注意： 答案中不可以包含重复的三元组。

例如,给定数组 nums = [-1, 0, 1, 2, -1, -4]，

满足要求的三元组集合为：
[
  [-1, 0, 1],
  [-1, -1, 2]
]

一、暴力法
三次for循环，求出所有满足条件的三个数字组合即可。
时间复杂度较高，O(n^3)

二、指针法
先固定一个数字，另外两个数字从头和尾依次遍历相加，找出符合条件的即可

C++代码如下：

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res; //定义一个二维数组用来存储结果
        sort(nums.begin(),nums.end());  //将nums数组进行排序
        int len = nums.size(); //计算数组长度
        if (len < 3 ||nums[0] >0 || nums[len - 1] < 0) return {};//判断数组是否符合条件
        for (int i = 0;i<len-1;i++){ //遍历数组中每一个数字
            int temp = nums[i];   // 用temp 存储每次遍历的那个数字，固定下来
            int l = i+1;         
            int r = len -1;       
             if(i>0 && temp==nums[i-1])    continue; //在遍历的过程中如果该数字与前一个数字相同则跳过
            while (l<r){
                if (nums[l]+nums[r] == -temp){
                       if (l == i+1 || r == len -1){
                           res.push_back(vector<int>{nums[i],nums[l],nums[r]});
                           l++;r--;
                       }
                        else if (nums[l] == nums[l-1]) l++;
                       else if(nums[r] == nums[r+1]) r--; //与上面continue 相同，避免重复
                     else{  
                         res.push_back(vector<int>{nums[i],nums[l],nums[r]});
                         l++;r--;
                     }
                    }
                
                else if (nums[l]+nums[r] < -temp) l++;
                else r--;
            }
        }
        return res;
    }
};