15. 三数之和

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/chen158_7/article/details/86647808

题目描述请点击查看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
展开阅读全文

没有更多推荐了,返回首页