15、三数之和(python)

题目:

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

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

示例:

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

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

代码(排序+双指针):

class Solution:
    def threeSum(self, nums):
        if len(nums) < 3:                 # 若数组长度 < 3,直接返回空
            return []
        nums = sorted(nums)               # 排序
        re = []                           # 结果变量
        for i in range(len(nums)):                  # 遍历
            if nums[i] > 0:                         # 若 nums[i] > 0,则没有数字之和为0,直接返回re
                return re 
            if i > 0 and nums[i] == nums[i-1]:      # 排除重复数组
                continue
            L = i + 1                               # 左指针
            R = len(nums) - 1                       # 右指针
            while L < R:                                      # 遍历
                temp = nums[i] + nums[L] + nums[R]            # 三数之和
                if temp == 0:                                 # 若temp = 0
                    re.append([nums[i],nums[L],nums[R]])      # 添加到结果列表中
                    while L < R and nums[L] == nums[L+1]:     # 避免左指针下一个数字重复,跳过
                        L = L + 1
                    while L < R and nums[R] == nums[R-1]:     # 避免右指针下一个数字重复,跳过
                        R = R - 1
                    L = L + 1                       # 左指针右移,遍历
                    R = R - 1                       # 右指针左移,遍历
                elif temp > 0:                                # 若temp > 0
                    R = R - 1                                 # 则应该让三数之和小一些,故右指针左移
                else:                                         
                    L = L + 1                                 # 反之,应该让三数之和大一些,故左指针右移
        return re
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值