【Hot100】15.三数之和

解法:排序 + 双指针

  • 首先对数组排序,便于后面处理重复元素。
  • 第一层循环遍历数组中的每一个元素,作为三元组中的第一个元素 nums[i] ,并跳过重复的元素。
  • 对于每个 i ,使用双指针 l (初始为 i+1)和 r (初始为数组末尾),在 i 之后的区间寻找满足 nums[l]+nums[r] 等于 -nums[i]。
  • 根据双指针指向的两数之和与目标值( -nums[i] )的大小关系:如果相等,将对应的三元组添加到结果数组中,移动指针,并跳过重复元素;如果小于target,左指针 l 右移;如果大于target,右指针 r 左移。
Go 复制代码
func threeSum(nums []int) [][]int {
    sort.Ints(nums)
    result := [][]int{}
    length:=len(nums)
    for i := 0;i<length;i++{
        //跳过重复元素
        if i>0&&nums[i]==nums[i-1]{
            continue
        }
        target:=-nums[i]
        l,r:=i+1,length-1
        for l<r {
            sum := nums[l]+nums[r]
            if sum == target {
                result=append(result,[]int{nums[i],nums[l],nums[r]})
                l++
                r--
                //跳过重复元素
                for l<r && nums[l]==nums[l-1]{ l++ }
                for l<r && nums[r] == nums[r+1]{ r-- }
            }else if  sum<target {
                l++
            }else{
                r--
            }
        }
    }
    return result
}

时间复杂度:排序 O(nlogn),循环 O(n2),总体 O(n2)。

空间复杂度:排序 O(logn),result数组排序 O(n2),总体排序 O(n2)。


我一开始的想法:

  • 首先对数组排序,便于后面处理重复元素。
  • 第一层循环遍历数组中的每一个元素,作为三元组中的第一个元素 nums[i] ,并跳过重复的元素。
  • 第二层循环遍历 i 之后的元素作为三元组的第二个数 nums[j],同样跳过重复的元素。
  • 第三层循环遍历 j 之后的元素,找到满足条件的第三个数 nums[m],将对应的三元组添加到结果 result 中,并跳出内层循环。
Go 复制代码
func threeSum(nums []int) [][]int {
    sort.Ints(nums)
    result := [][]int{}
    length:=len(nums)
//解法一
    for i:=0;i<length;i++{
        if i>0 && nums[i] == nums[i-1]{
            continue
        }
        n := -nums[i]
        for j:=i+1;j<length;j++{
            if j>i+1 && nums[j]==nums[j-1]{
                continue
            }
            k:= n-nums[j]
            for index,v := range nums[j+1:] {
                if v == k {
                    m := index+j+1
                    result=append(result,[]int{nums[i],nums[j],nums[m]})
                    break
                }
            }
        }
    }
    return result
}

时间复杂度:排序 O(nlogn),循环O(n3),总体O(n3)。

空间复杂度:排序 O(logn),存储结果的result二维数组O(n2),总体O(n2)

相关推荐
千金裘换酒5 小时前
LeetCode 移动零元素 快慢指针
算法·leetcode·职场和发展
wm10435 小时前
机器学习第二讲 KNN算法
人工智能·算法·机器学习
NAGNIP5 小时前
一文搞懂机器学习线性代数基础知识!
算法
NAGNIP6 小时前
机器学习入门概述一览
算法
iuu_star6 小时前
C语言数据结构-顺序查找、折半查找
c语言·数据结构·算法
Yzzz-F6 小时前
P1558 色板游戏 [线段树 + 二进制状态压缩 + 懒标记区间重置]
算法
漫随流水6 小时前
leetcode算法(515.在每个树行中找最大值)
数据结构·算法·leetcode·二叉树
mit6.8247 小时前
dfs|前后缀分解
算法
扫地的小何尚7 小时前
NVIDIA RTX PC开源AI工具升级:加速LLM和扩散模型的性能革命
人工智能·python·算法·开源·nvidia·1024程序员节
千金裘换酒8 小时前
LeetCode反转链表
算法·leetcode·链表