leetcode15. 三数之和

最新推荐文章于 2024-08-27 14:15:34 发布

一条特立独行的狗、

最新推荐文章于 2024-08-27 14:15:34 发布

阅读量104

点赞数 1

分类专栏： Leetcode数组文章标签： leetcode 数组

本文链接：https://blog.csdn.net/weixin_43498985/article/details/86673237

版权

Leetcode数组专栏收录该内容

68 篇文章 0 订阅

订阅专栏

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

例如, 给定数组 nums = [-1, 0, 1, 2, -1, -4]，
满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]

思路：先排个序，将数组头部nums[i]定为第一个数，nums[j]为它后面的数，定为第二个数，再将数组最后一个数nums[k]定为第三个数，这样就得到三个数的和，如果大于0，则j需要后移，如果小于0，则k前移，等于0的话就符合，压入vector中，然后再看此时j的后面有没有和nums[j]相等的数，有就排除掉（移动j），最后再看i后面有没有和nums[i]相等的数，有的话一样要排除掉，为了防止找到重复的三元组，最后得到的vecRes就包含了所有的结果。

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>> vecRes;
        int nCount = nums.size();
        if(nCount < 3)
            return vecRes;
        sort(nums.begin(),nums.end());
        int i=0,j,k;
        while(i < nCount && nums[i] <= 0)
        {
            j = i+1;
            k = nCount-1;
            while(j<k)
            {
                int nSum = nums[i] + nums[j] + nums[k];
                if(nSum < 0)
                {
                    j++;
                    continue;
                }
                else if(nSum > 0)
                {
                    k--;
                    continue;
                }
                vecRes.push_back({nums[i],nums[j],nums[k]});
                while(j+1<k && nums[j] == nums[j+1])
                    j++;
                j++;
            }
            while(i+1<nCount && nums[i] == nums[i+1])
                i++;
            i++;
        }
        return vecRes;
    }
};