【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)

相关推荐
弈宸2 天前
Transformer与ViT
算法·架构
程序员爱钓鱼2 天前
Go语言实战案例-项目实战篇:使用Go调用第三方API(如天气、翻译)
后端·google·go
知其然亦知其所以然3 天前
国产大模型也能无缝接入!Spring AI + 智谱 AI 实战指南
java·后端·算法
然我3 天前
搞定异步任务依赖:Promise.all 与拓扑排序的妙用
前端·javascript·算法
徐小夕3 天前
支持1000+用户同时在线的AI多人协同文档JitWord,深度剖析
前端·vue.js·算法
程序员爱钓鱼3 天前
Go语言实战案例-项目实战篇:开发一个 IP 归属地查询接口
后端·google·go
沐怡旸4 天前
【算法】【链表】328.奇偶链表--通俗讲解
算法·面试
Java水解4 天前
go语言教程(全网最全,持续更新补全)
后端·go
掘金安东尼4 天前
Amazon Lambda + API Gateway 实战,无服务器架构入门
算法·架构
码流之上4 天前
【一看就会一写就废 指间算法】设计电子表格 —— 哈希表、字符串处理
javascript·算法