LeetCode 15.三数之和

84 阅读1分钟

从零开始学习c++,每天起码做一道leetcode题目,在此记录,希望最后能够有所收获!

一、题目描述

15.三数之和

给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

示例1:

输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]
解释:
nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。
nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。
nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。
不同的三元组是 [-1,0,1][-1,-1,2] 。
注意,输出的顺序和三元组的顺序并不重要。

示例 2:

输入: nums = [0,1,1]
输出: []
解释: 唯一可能的三元组和不为 0 。

示例 3:

输入: nums = [0,0,0]
输出: [[0,0,0]]
解释: 唯一可能的三元组和为 0

提示:

  • 3 <= nums.length <= 3000
  • -105 <= nums[i] <= 105

二、思路分析

最难想到的就是a如何进行去重,看代码里关于a去重的部分,之所以可以continue得原因是因为前面结果集已经有了对应元素,比如当前是-1,已经把能凑到0的结果全搞出来了,当你下一个元素还是-1.结果集都是一样的就去重了。也就是说i元素时遍历了后面的所有可能性,所以后面再出现和i相等的元素时,必然是重复的。

三、AC代码

class Solution {
public:
    vector<vector<int>> threeSum(vector<int>& nums) {
        vector<vector<int>>ans;
        sort(nums.begin(),nums.end());
        
        int right=nums.size()-1;
        for(int i=0;i<nums.size();i++){
            if(nums[i]>0){
                return ans;
            }
            if(i>0&&nums[i]==nums[i-1]){
                continue;
            }

            int left=i+1;
            int right=nums.size()-1;
            while(right>left){
                if(nums[i]+nums[left]+nums[right]>0){
                    right--;
                }
                else if(nums[i]+nums[left]+nums[right]<0){
                    left++;

                }
                else{
                    ans.push_back({nums[i],nums[left],nums[right]});
                    while(right>left&&nums[left]==nums[left+1]){
                        left++;
                    }
                    while(right>left&&nums[right]==nums[right-1]){
                        right--;
                    }
                    left++;
                    right--;
                }
            }      

        }
        return ans;

    }
};

四、总结

难度比较大