首页 > 编程语言 > 详细

leetcode-三数之和python

时间:2020-03-15 10:48:16      阅读:63      评论:0      收藏:0      [点我收藏+]

给你一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a,b,c ,使得 a + b + c = 0 ?请你找出所有满足条件且不重复的三元组。

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

示例:

给定数组 nums = [-1, 0, 1, 2, -1, -4],

满足要求的三元组集合为:
[
  [-1, 0, 1],
  [-1, -1, 2]
]

思路:先对数组进行排序,然后使用双指针,左右逼近,遍历数组作为外层循环,重点是去重,left或right碰到相同数字则往下跳一个

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        #-4,-1,-1,0,1,2
        #第一步,判断边界
        res = []
        nums.sort()
        n=len(nums)
        for i in range(n-2):
            if i==0 or nums[i]>nums[i-1]:#去重    
                left,right = i+1,n-1
                while left<right:
                    ident = nums[i]+nums[left]+nums[right]
                    if ident==0:
                        res.append([nums[i],nums[left],nums[right]])
                        left+=1
                        right-=1
                        #去重
                        while left<right and nums[left]==nums[left-1]:
                            left+=1
                        while left<right and nums[right]==nums[right+1]:
                            right-=1
                    elif ident<0:
                        left+=1
                    else:
                        right-=1
        return res

 

leetcode-三数之和python

原文:https://www.cnblogs.com/xiao-longxia/p/12496290.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!