欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-04-02 22:55:28
将两个有序链表合并为一个新的有序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。 注意: 两个链表都是有序的 如果某个链表为空,那就直接返回另外一个有序链表 然后我们就要比较两个链表的节点的顺序了 首先,我们定义一个result指针 比较两个链表的第一个元素哪个比较小 result指向小 ......

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

输入: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指向小的那个链表
先来看一张图
合并两个有序链表的golang实现

核心代码:

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接上那个元素