15 三数之和(三指针)

1. 问题描述:

给你一个包含 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]
输出:[]

提示:

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/3sum/

2. 思路分析:

这道题目是一道非常经典的问题,因为数据规模为1000,所以时间复杂度只能够控制在O(n ^ 2)或者O(n ^ 2logn)以内,所以我们需要考虑一下如何来枚举,为了方便后面的处理我们可以先对数组排序,最容易想到的是三重循环来枚举但是我们需要对三重循环优化一下才可以,通过分析发现如果使用三重循环来枚举,重复的三元组为当前枚举的数字不是第一个数字并且与上一个数字相同的情况,所以当我们发现这种情况的时候直接跳过当前枚举的数字即可,因为排序之后往后的数字是越来越大的所以一开始的时候我们令第三个指针k = n - 1,对于第二个指针j往后走的时候,指针k是单调往左走的,所以两个指针是具有单调性,那么我们就可以使用双指针来解决,通过三个指针就可以求解出所有不重复的三元组,时间复杂度低于O(n ^ 3)。

3. 代码如下:

from typing import List


class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        res = list()
        nums.sort()
        n = len(nums)
        for i in range(n):
            # 注意k的位置是写在这里
            k = n - 1
            # 跳过重复的数字
            if i > 0 and nums[i] == nums[i - 1]: continue
            for j in range(i + 1, n):
                if j > i + 1 and nums[j] == nums[j - 1]: continue
                while j < k and nums[j] + nums[k] > -nums[i]: k -= 1
                if j == k:
                    break
                if nums[i] + nums[j] + nums[k] == 0:
                    res.append([nums[i], nums[j], nums[k]])
        return res
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值