数组中和为0的三个数

题目描述:
给定一个包含 n 个整数的数组 nums,判断 nums 中是否存在三个元素 a ,b ,c ,使得 a + b + c = 0 ?请找出所有和为 0 且 不重复 的三元组。

示例 1:
输入:nums = [-1,0,1,2,-1,-4]
输出:[[-1,-1,2],[-1,0,1]]

示例 2:
输入:nums = []
输出:[]

示例 3:
输入:nums = [0]
输出:[]

提示:
0 <= nums.length <= 3000
-105 <= nums[i] <= 105

解答思路:
如果一上来就3个循环暴力破解, 时间复杂度就为O(n3), 需要换个思路, 不妨先排个序, 将[-1,0,1,2,-1,-4] 变成 [-4, -1, -1, 0, 1, 2], 接下来, 先固定第一个数 -4, 然后在
[-1, -1, 0, 1, 2]这个列表中进行双指针遍历, 这个类似于我之前写的 排序数组中2个数字之和的3中解法

def double_point(nums):
    # 对列表进行升序排序
    nums.sort()
    result_list = list()
    for n, i in enumerate(nums):
        # 如果大于0了, 后续加起来肯定不会等于0
        if i > 0:
            break
        # 如果和上一个数一样, 那就没必要重复了
        if n > 0 and i == nums[n - 1]:
            continue
        # 对列表进行切片
        nums2 = nums[n + 1:]
        # 如果切片后的列表长度等于2,说明已经是最后3个了
        if len(nums2) == 2:
            if sum(nums[n:]) == 0:
                result_list.append(nums[n:])
            break
        # 开始双指针查找
        low, high = (0, len(nums2) - 1)
        while low < high:
            if i + nums2[low] + nums2[high] == 0:
                result_list.append([i, nums2[low], nums2[high]])
                while True:
                    # try是为了防止报错
                    try:
                        # 如果下一个也是一样的数,那就不必要重复了
                        if nums2[low] == nums2[low + 1]:
                            low += 1
                        else:
                            break
                    except:
                        break
                # 下一个不是一样的数, 那就把左边的指针往右边移动
                low += 1
            # 如果3个数加起来小于0, 则把指针往右移动
            elif i + nums2[low] + nums2[high] < 0:
                low += 1
            else:
                high -= 1
    return result_list

最后在力扣上面跑的分数如下, 速度比官方解答还快几十ms
在这里插入图片描述
如果对您有帮助, 麻烦您点个赞, 如果有问题或不同的想法欢迎交流, 一起进步

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值