合并两个有序链表的golang实现

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

输入:1->2->4, 1->3->4
输出:1->1->2->3->4->4

注意:

  • 两个链表都是有序的
  • 如果某个链表为空,那就直接返回另外一个有序链表
//如果有一条链是nil,直接返回另外一条链
if l1 == nil {
    return l2
}
if l2 == nil {
    return l1
}

然后我们就要比较两个链表的节点的顺序了

  • 首先,我们定义一个result指针
  • 比较两个链表的第一个元素哪个比较小
  • result指向小的那个链表
先来看一张图

核心代码:

func mergeTwoLists(l1 *ListNode, l2 *ListNode) *ListNode {
    //如果有一条链是nil,直接返回另外一条链
    if l1 == nil {
        return l2
    }
    if l2 == nil {
        return l1
    }
    // 定义一个结果节点
    var res *ListNode
    // 当l1节点的值大于l2节点的值,那么res指向l2的节点,从l2开始遍历,反之从l1开始
    if l1.Val >= l2.Val {
        res = l2
        res.Next = mergeTwoLists(l1, l2.Next)
    } else {
        res = l1
        res.Next = mergeTwoLists(l1.Next, l2)
    }
    return res
}

使用递归,不断去找两个链表中比较小的元素,然后result接上那个元素

posted @ 2018-12-18 21:28  timliudream  阅读(1874)  评论(0编辑  收藏  举报