leetcode 15 三数之和

题目详情

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

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

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

满足要求的三元组集合为：
[
[-1, 0, 1],
[-1, -1, 2]
]
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum

题解

1、首先对nums排序
2、寻找三元组的思路是，先确定一个数nums[k]，然后在nums剩下的数里寻找和等于-nums[k]的两个数
3、由于三个数的和要求为0，如果三个数不同时为0，则三个数里至少有一个负数，所以nums[k]只需考虑负数，令k从0开始，直到nums[k]>0时结束循环
4、对于nums[k]，接下来的任务是从nums[k+1]到nums[size-1]里寻找两个和等于target即-nums[k]的数，设立两个指针i，j，i初始化为k+1，j初始化为size-1，计算nums[i]+nums[j]的值，如果nums[i]+nums[j]<target，则i++，如果nums[i]+nums[j]>target，则j–，直到i==j
5、如果nums[i]+nums[j]等于target，那么记下这个三元组，然后将i和j均更新，并且跳过重复的数值。

代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        sort(nums.begin(),nums.end());
        
        vector<vector<int>> result;
        if(nums.size()<3||nums[0]>0||nums.back()<0)
            return result;
        
        for(int i=0;i<nums.size();i++)
        {
            if(nums[i]>0)
                break;
            if(i>0&&nums[i]==nums[i-1])
                continue;
            
            int p,q,target;     
            target=-nums[i];p=i+1;q=nums.size()-1;  
            while(p<q)
            {
                if((nums[p]+nums[q])==target)
                {
                    result.push_back({nums[i],nums[p],nums[q]});
                    do
                    {
                        p++;
                    }while(nums[p]==nums[p-1]&&p<q);
                    do
                    {
                        q--;
                    }while(nums[q]==nums[q+1]&&p<q);
                }
                else if((nums[p]+nums[q])>target)
                    q--;
                else
                    p++;
            }
        }
        return result;
    }
};