leetcode—python 50天刷题第9题三数之和

最新推荐文章于 2023-10-07 19:48:45 发布

weixin_45556991

最新推荐文章于 2023-10-07 19:48:45 发布

阅读量82

点赞数

分类专栏： 50天刷题文章标签： leetcode python 双指针法

本文链接：https://blog.csdn.net/weixin_45556991/article/details/100165850

版权

50天刷题专栏收录该内容

51 篇文章 0 订阅

订阅专栏

题目描述

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

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/3sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解答过程

class Solution(object):
    def threeSum(self, nums):
        """
        :type nums: List[int]
        :rtype: List[List[int]]
        """
        nums.sort()
        a=len(nums)
        m=0
        final=[]
        for i in range(a-2):
            if nums[i]>0:
                break
            if i>0 and nums[i]==nums[i-1]:continue
                
                    
            if m==1:
                break
            c=i+1
            d=a-1
            while c<d:
                if (nums[i]+nums[c]+nums[d])==0:
                    final.append([nums[i],nums[c],nums[d]])
                    c=c+1
                    d=d-1
                    if c>=d:
                        continue
                    while c<d and nums[c]==nums[c-1]:
                        c=c+1
                    while c<d and nums[d]==nums[d+1]:
                        d=d-1
                if nums[i]+nums[c]+nums[d]<0:
                    c=c+1                    
                    while  c<d and nums[c]==nums[c-1]:
                        c=c+1
                if nums[i]+nums[c]+nums[d]>0:
                    d=d-1                    
                    while  c<d and nums[d]==nums[d+1]:
                        d=d-1  
        return final