15. 三数之和

题目描述请点击查看LeetCode 题目描述

Python3 代码解答如下:

class Solution:
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        """
        思路：先将数组排序，后循环数组，假定第 i 个数为第一个元素，第二个元素为第i+1个，第3个元素为数组最后一个，因为前面的都是数值较小的，要想找到与其相反的大数，从后面开始会比较快，因为不能重复，所以要保证第二个元素下标要小于第3个元素
        """
        nums.sort()
        res =[]
        i = 0
        for i in range(len(nums)):
            #当第一次循环的第一个元素与第二次循环的第一个元素相等时，不必找了，即使找到也都是重复的了
            if i == 0 or nums[i]>nums[i-1]:
                l = i+1
                r = len(nums)-1
                #限制第二个元素与第三个元素的重复使用，比如2,3 = 3,2
                while l < r:
                    s = nums[i] + nums[l] +nums[r]
                    if s ==0:
                        #找到元素，追加到列表中，并第二个元素加1，第三个元素减1，两个只要有一个不变，即使有满足条件的元素，也会重复
                        res.append([nums[i],nums[l],nums[r]])
                        l +=1
                        r -=1
                        #判断下标+1后的第二个元素与第二个元素是否一样，如果一样依然是+1
                        while l < r and nums[l] == nums[l-1]:
                            l += 1
                        #判断下标-1后的第三个元素与第三个元素是否一样，如果一样依然是-1
                        while r > l and nums[r] == nums[r+1]:
                            r -= 1
                    elif s>0:
                        #当和大于0时，说明第三个元素的值太大，需要一个稍小一点的，所有下标-1
                        r -=1
                    else :
                        #当和小于0时，说明前两个元素的值太小，需要一个稍大一点的，因为第一个元素固定为第 i个，所以第二元素下标+1
                        l +=1
        return res