【LeetCode】15. 三数之和

不念不忘少年蓝@ 2022-05-08 02:26 179阅读 0赞

题目链接：[https://leetcode-cn.com/problems/3sum/description/][https_leetcode-cn.com_problems_3sum_description]

**题目描述**

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

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

**示例**

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

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

**解决方法**

首先从小到大排序，然后遍历数组中每一个元素，用双指针找到另外两个元素，将三元组存入vector中（注意在此过程中要去重）

class Solution { 
    public:
        vector<vector<int>> threeSum(vector<int>& nums) { 
    // 首先从小到大排序，然后遍历数组中每一个元素，用双指针找到另外两个元素,将三元组存入vector中（注意在此过程中要去重）
            
            vector<vector<int>> result;
            // 1.排序
            sort(nums.begin(),nums.end());
            
            // 2.找到三元组
            for (int i=0;i<nums.size();i++){ 
                if (nums[i]>0) break; //剪枝操作，如果当前数大于0则break
                if (i>0 && nums[i]==nums[i-1]) continue; //去重操作
                
                int left=i+1, right=nums.size()-1;
                while(left<right){ 
                    int sum=nums[i]+nums[left]+nums[right];
                    if (sum>0) right--; 
                    else if (sum<0) left++;
                    else{ 
                        result.push_back({ nums[i],nums[left],nums[right]});
                        while (left<right && nums[left]==nums[left+1]) left++; //这两条语句是为了去重
                        while (left<right && nums[right]==nums[right-1]) right--;
                        left++;right--;
                    }
                }
                
            }
    
            return result;
        }
    };

[https_leetcode-cn.com_problems_3sum_description]: https://leetcode-cn.com/problems/3sum/description/