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)$$
  这里的n和m分别为两个链表的长度
• 空间复杂度为：
  $$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

}