LeetCode--15. 三数之和（3Sum）

最新推荐文章于 2022-11-14 17:23:54 发布

Algo233

最新推荐文章于 2022-11-14 17:23:54 发布

阅读量102

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq_37746778/article/details/88549696

版权

LeetCode 专栏收录该内容

11 篇文章 0 订阅

订阅专栏

题目描述

给定一个包含 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>> ans;
        if(nums.size()<3)
            return ans;
        sort(nums.begin(),nums.end());//先对num进行排序
        //由于三元组和为0的数，除了（0,0,0），一定会有一个负数
        //只需遍历负数和0即可得出全部结果(不用再遍历正数)
        //然后再用两个指针，一个从左至右，一个从右至左，
        //若遇到重复的数据需要跳过，直至和为0
        int i=0;
        int j,k;//左右指针
        int l=nums.size()-1;
        while(nums[i]<=0&&i<=l-2)//小于三个数时不再遍历
        {           
            if(i>0&&nums[i]==nums[i-1])//防止首数重复
            {
                ++i;
                continue;       
            }    
            j=i+1;k=l;
            while(j<k)
            {
               
                if(nums[j]+nums[k]+nums[i]==0)
                {
                    vector<int> temp{nums[i]};//先将负数入队 
                    temp.push_back(nums[j]);
                    temp.push_back(nums[k]);
                    ans.push_back(temp);
                    ++j;--k;
                    //跳过重复数字，防止重复数组
                    while(j<k&&nums[j]==nums[j-1])          
                        ++j;
                    while(j<k&&nums[k]==nums[k+1])           
                        --k;
                    
                }
                else if(nums[j]+nums[k]+nums[i]<0)
                {//小于0情况
                    ++j;
                }
                else
                {//大于0情况
                    --k;
                }
                
                
            }           
            ++i;
        }
        return ans;
    }
};