leetcode三数之和C++版

题目描述:

给定一个包含 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;
    }
};

你可能感兴趣的:(c++)