【LeetCode】15. 三数之和

1 问题

给你一个整数数组 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 。

2 答案

自己写的，超出时间限制

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        res_list = []
        for i in range(len(nums)-2):
            for j in range(i+1, len(nums)-1):
                for k in range (j+1, len(nums)):
                    if nums[i]+nums[j]+nums[k] == 0 and i != j != k:
                        list1 = [nums[i], nums[j], nums[k]]
                        list1.sort()  # 没有返回值
                        # list1 = sorted([nums[i], nums[j], nums[k]])  # 也可以用这个
                        if list1 in res_list:
                            pass
                        else:
                            res_list.append(list1) 
        return res_list

官方解

双指针

算法设计上比较巧妙，可以减少很多不必要的计算

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        
        n=len(nums)
        res=[]
        if(not nums or n<3):
            return []
        nums.sort()
        res=[]
        for i in range(n):
            if(nums[i]>0):
                return res 
            if(i>0 and nums[i]==nums[i-1]):  # i重复元素跳过
                continue
            L=i+1
            R=n-1
            while(L<R):
                if(nums[i]+nums[L]+nums[R]==0):
                    res.append([nums[i],nums[L],nums[R]])
                    while(L<R and nums[L]==nums[L+1]):  # L重复元素跳过
                        L=L+1
                    while(L<R and nums[R]==nums[R-1]):  # R重复元素跳过
                        R=R-1
                    L=L+1
                    R=R-1
                elif(nums[i]+nums[L]+nums[R]>0):
                    R=R-1
                else:
                    L=L+1
        return res