Leetcode-三数之和

Leetcode-三数之和

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

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

方法一：暴力法

最先开始的思路当然是直接找咯- -，先将数组进行排序，然后三重循环数组，如果和为0，将他们丢入list，判断list在res里有没有，如果有就不管，没有的话append进res

class Solution:
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        new_nums = nums
        new_nums.sort()
        length = len(nums)
        res = []
        for i in range(length - 2):
            for j in range(i+1, length -1):
                for k in range(j+1, length):
                    if new_nums[i]+new_nums[j]+new_nums[k] == 0:
                        temp = [new_nums[i], new_nums[j] ,new_nums[k]]
                        if temp not in res:
                            res.append(temp)                
        return res

• 然后，果不其然，超出时间限制- -

方法二：想着三数和为0，那其实就是三种情况：+±、–+和000（0算正数），然后排序后找这三个数，但结果仍然超出时间限制，忧伤- -

方法三：排序，然后循环数组，通过判断余下两个数其中一个在中间另一个在右边，判断三数相加是否为0 ，然后更新后面两个数

class Solution:
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        res = []
        length = len(nums)
        nums.sort()  
        for i in range(length): 
            l = i+1
            r = length-1            
            while(l < r):
                if nums[i] + nums[l] + nums[r] == 0:                        
                    temp = [nums[i],nums[l],nums[r]]
                    if temp not in res:
                        res.append(temp)
                    r -= 1
                    l += 1                   
                elif nums[i] + nums[l] + nums[r] > 0:
                    r -= 1
                else:
                    l += 1

        return res