leetcode热题100.三数之和

简介: leetcode热题100.三数之和

题目

给你一个整数数组 nums ,判断是否存在三元组 [nums[i], nums[j], nums[k]] 满足 i != j、i != k 且 j != k ,同时还满足 nums[i] + nums[j] + nums[k] == 0 。请

你返回所有和为 0 且不重复的三元组。

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

示例 1:

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

输出:[[-1,-1,2],[-1,0,1]]

解释:

nums[0] + nums[1] + nums[2] = (-1) + 0 + 1 = 0 。

nums[1] + nums[2] + nums[4] = 0 + 1 + (-1) = 0 。

nums[0] + nums[3] + nums[4] = (-1) + 2 + (-1) = 0 。

不同的三元组是 [-1,0,1] 和 [-1,-1,2] 。

注意,输出的顺序和三元组的顺序并不重要。

示例 2:

输入:nums = [0,1,1]

输出:[]

解释:唯一可能的三元组和不为 0 。

示例 3:

输入:nums = [0,0,0]

输出:[[0,0,0]]

解释:唯一可能的三元组和为 0 。

提示:

3 <= nums.length <= 3000

− 1 0 5 -10^5105 <= nums[i] <= 1 0 5 10^5105

解题方法

我们先考虑最朴实的解法,我们可以遍历三次数组,计算这三次的和,判断是否为零,再通过集合去重,但是这样的时间复杂度是 o ( n 3 ) o(n^3)o(n3),我们思考一下是否可以进行优化。

我们在上面的算法中计算了很多重复的值,我们可以尝试以下排序整个数组,这样我们第一次遍历结束之后,第二次遍历就不需要考虑第一次遍历的数左边的数了,实现了一次剪枝;对于第三次遍历也是一样的,我们不需要考虑第二次遍历的数左边的数。

我们考虑第二,第三次遍历,发现如果此时 nums[i] + nums[j] + nums[k] < 0, 因为 i 0,我们期望他变小一点,则左移 k,检查是否可以等于0。

如果等于零了,说明我们找到了答案,加入最后的结果中即可

注意:关于结果的去重,对于i,left,right,我们都使用while循环排除重复的值,保证结果的唯一性

复杂度

时间复杂度:

遍历了两次数组,所以是: O ( n 2 ) O(n^2)O(n2)

空间复杂度:

添加空间复杂度, 示例: O ( n ) O(n)O(n)

Code

class Solution:
    def threeSum(self, nums: List[int]) -> List[List[int]]:
        nums = sorted(nums)
        n = len(nums)
        ans = []
        i = 0 
        while i < n-2:
            while i<n and i>0 and nums[i] == nums[i-1]:
                i+=1
            
            if i < n-2 and nums[i+1] + nums[i+2] + nums[i]>0:
                break
            
            if i < n-2 and nums[i] + nums[-1] + nums[-2] <0:
                i+=1
                continue
            left,right = i+1,n-1
            while left<right:
                if nums[left]+nums[right] + nums[i] < 0:
                    left += 1
                elif  nums[left]+nums[right] + nums[i] > 0:
                    right -= 1
                else:
                    ans.append([nums[i],nums[left],nums[right]])
                    left+=1
                    while left<right and nums[left]==nums[left-1]:
                        left+=1
                    right-=1
                    while left<right and nums[right]==nums[right+1]:
                        right-=1
            i+=1
        
        return ans


目录
相关文章
|
6月前
LeetCode 热题100——单调栈
LeetCode 热题100——单调栈
57 0
|
6月前
|
存储
LeetCode热题 首题 两数之和
LeetCode热题 首题 两数之和
39 1
|
11月前
LeetCode 热题100——栈与队列专题(三)
LeetCode 热题100——栈与队列专题(三)
46 1
|
6月前
|
vr&ar
leetcode热题100.路径总和 III
leetcode热题100.路径总和 III
40 1
|
6月前
|
机器学习/深度学习 算法 索引
leetcode热题100.两数之和
leetcode热题100.两数之和
36 1
|
6月前
|
存储 算法
《LeetCode 热题 HOT 100》——寻找两个正序数组的中位数
《LeetCode 热题 HOT 100》——寻找两个正序数组的中位数
|
6月前
|
网络协议
《 LeetCode 热题 HOT 100》——无重复字符的最长子串
《 LeetCode 热题 HOT 100》——无重复字符的最长子串
|
6月前
《LeetCode 热题 HOT 100》—— 两数相加
《LeetCode 热题 HOT 100》—— 两数相加
|
6月前
leetcode热题100. 字母异位词分组
leetcode热题100. 字母异位词分组
46 0
|
6月前
leetcode热题100.二叉树中的最大路径和
leetcode热题100.二叉树中的最大路径和
35 0