每日一题20201109(15. 三数之和)

简介: 三数之和解法

题目链接:  15. 三数之和


10.jpg

image.png

解法


  • 暴力法


首先可以确定的是暴力法的时间复杂度是O(N³), 所以基本上不考虑此等解法。
  • 排序+双指针


思路是先将数组排序(从小到大),然后固定数组的第一位。
定义2个指针(左右指针)分别指向定位数组的后一位和数组最后一位。
如果3个数字加起来比0小,则左指针右移,循环继续。
如果3个数字加起来比0大,则右指针左移,循环继续。
如果3个数字等于0,说明找到了结果,将3个数字放入结果数组中并把左指针右移,右指针左移。
1. 这里需要注意的是,可能会有重复数据产生,为了不产生重复数据,需要确保左移/右移后的值与之前不一致。
2. 如果从固定的数字大于0了,说明右侧不可能有结果了,因为右侧都是大于0的数字,加起来不会大于0了。
3. 如果固定位置与上一个固定位置的值相同,也会产生重复数据,遇到直接跳过就行。

网络异常,图片无法展示
|

image.png

最终代码



class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        # 先对数组进行排序, 为了方便比较元素
        nums = sorted(nums)
        i = 0  # 数组起点
        result = []    # 结果
        while i < len(nums):
            if nums[i] > 0:
                # 大于0的数字可以直接跳过
                break
            if i > 0 and nums[i] == nums[i-1]:
                # 与上一个固定位重复的数字直接跳过
                i += 1
                continue 
            front, back = i+1, len(nums)-1
            while front < back:
                val = nums[i] + nums[back] + nums[front]
                if val < 0:
                    # 说明需要增大数字, front+1
                    front += 1
                elif val > 0:
                    # 说明需要缩小数字, back-1
                    back -= 1
                else:
                    result.append([nums[i], nums[back], nums[front]])
                    front += 1
                    back -= 1
                    # 保证没有重复数组
                    while front < len(nums) and nums[front] == nums[front-1]:
                        front += 1
                    while back >= 0 and nums[back] == nums[back+1]:
                        back -= 1
            i += 1
        return result



相关文章
|
1月前
|
存储
Leetcode第十五题(三数之和)
LeetCode第十五题“三数之和”要求在一个整数数组中找出所有不重复的三元组,使得它们的和为0,通常通过先排序再使用双指针法来解决。
38 0
Leetcode第十五题(三数之和)
|
5月前
|
算法
OJ刷题:杨氏矩阵
OJ刷题:杨氏矩阵
27 0
|
5月前
|
算法 容器
【LeetCode刷题】三数之和、四数之和
【LeetCode刷题】三数之和、四数之和
|
6月前
|
算法
leetcode热题100.三数之和
leetcode热题100.三数之和
37 1
|
存储
每日一题——三数之和(双指针)
每日一题——三数之和(双指针)
每日一题——四数之和(双指针解法)
每日一题——四数之和(双指针解法)
Leetcode-每日一题927. 三等分(双指针)
题目需要你帮他把这个arr序列分成连续的三段序列,序列是由0、1组成,每段的0、1序列组成的二进制数都要相等。你只需要找到第一段序列与第二段序列的分割点 i, 第二段序列与第三段序列的分割点j。
94 0
Leetcode-每日一题927. 三等分(双指针)