Leetcode21. 合并两个有序链表(golang)

一、题目

题目来源:https://leetcode-cn.com/problems/merge-two-sorted-lists/

将两个升序链表合并为一个新的 升序 链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。

  • 示例 1:
    在这里插入图片描述

输入:l1 = [1,2,4], l2 = [1,3,4]
输出:[1,1,2,3,4,4]

  • 示例 2:

输入:l1 = [], l2 = []
输出:[]

  • 示例 3:

输入:l1 = [], l2 = [0]
输出:[0]

  • 提示:

两个链表的节点数目范围是 [0, 50]
-100 <= Node.val <= 100
l1 和 l2 均按 非递减顺序 排列

二、题解

定义一个新结点cur,将两个链表的值依次对比,每次将值小的节点连接至新结点后。最后返回cur即为所求结点。特殊情况:若链表长度不同,那么短链表遍历完而长链表一定没有遍历完,根据题目可知链表本来就是有序的,因此当短链表的节点插入完毕后,长链表剩余的节点值一定比新链表最后一个节点的值大,直接将整个剩余的长链表连接在新链表最后即可。

具体思想见笔者图示与代码注释

图示如下:

在这里插入图片描述

  • 时间复杂度为:
    0 ( n + m ) 0(n+m) 0(n+m)
    这里的n和m分别为两个链表的长度
  • 空间复杂度为:
    0 ( 1 ) 0(1) 0(1)

代码

func mergeTwoLists(l1 *ListNode, l2 *ListNode) *ListNode {
    
    
    //三种特殊情况,都为空返回空,一个为空则返回另一个
    if l1 == nil && l2 == nil{
    
    
        return nil
    }
    if l1 == nil && l2 != nil{
    
    
        return l2
    }
    if l2 == nil && l1 != nil{
    
    
        return l1
    }
    //定义pre用来连接大小比对完毕较小的那个节点
    pre := &ListNode{
    
    }
    //cur用来返回最后连接好的链表
    cur := pre
    //遍历两个链表
    for l1 != nil && l2 != nil{
    
    
        //找出值较小的那个节点,连接至pre
        if l1.Val < l2.Val{
    
    
            pre.Next = l1
            l1 = l1.Next
        }else{
    
    
            pre.Next = l2
            l2 = l2.Next
        }
        pre = pre.Next
    }
    //一个链表遍历完而另一个没遍历完,直接将仍有节点的链表插入尾部
    if l1 != nil{
    
    
        pre.Next = l1
    }
    if l2 != nil{
    
    
        pre.Next = l2
    }
    return cur.Next

}

猜你喜欢

转载自blog.csdn.net/QQ395879676/article/details/115771900
今日推荐