leetcode 15.三数之和

Posted 2021-02-11 bejm

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

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

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

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

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> res;
        sort(nums.begin(),nums.end());                       //对于数组进行升序排序
        if(!nums.empty() && nums.back() < 0) return {};      //如果数组nums不为空且数组的最后一项小于0，直接返回空
        for(int i = 0; i < nums.size(); i++){
            if(nums[i] > 0) break;                           //如果数组开始第一个数字大于0，那么后面的数字都是大于0,直接循环停止
            if(i > 0 && nums[i] == nums[i-1]) continue;      //如果数组前一个和后一个相等的话，直接跳过后一个，防止重复
            int target = 0 - nums[i];       
            int m = i + 1;                                   //左指针为m
            int n = nums.size() - 1;                         //右指针为n
            while(m < n){
                if(nums[m] + nums[n] == target){
                  res.push_back({nums[i],nums[m],nums[n]});  //如果nums[m]+nums[n]=target,直接记录三个数字的值
                  while(m < n && nums[m] == nums[m+1]) m++;  //如果m<n且后一个和前一个数字相等，那么m++
                  while(m < n && nums[n] == nums[n-1]) n--;  //如果m<n且后一个和前一个数字相等，那么n--
                  m++;                                       //如果以上条件不满足，直接将m++,n--
                  n--; 
                }
                else if(nums[m] + nums[n] < target) m++;     //如果两数相加之和小于target,则将左指针m++
                else n--;                                    //如果两数相加之和大于target,则将右指针n--
            } 
                   
        }
        return res;       
    }
};