【腾讯精选练习50题】21. 合并两个有序链表（简单）

题目描述

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

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

示例1

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

输出：[1,1,2,3,4,4]

示例2

输入：l1 = [], l2 = []

输出：[]

示例3

输入：l1 = [], l2 = [0]

输出：[0]

思路分析

• 这个就比较简单了，依次比较ListNode头大小即可
• 看了标签和评论，发现还有一种递归的做法，也试着写了一下

AC代码

迭代

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
// 执行用时：0 ms, 在所有 Go 提交中击败了100.00%的用户
// 内存消耗：2.5 MB, 在所有 Go 提交中击败了100.00%的用户
func mergeTwoLists(l1 *ListNode, l2 *ListNode) *ListNode {
    if l1 == nil {
        return l2
    }
    if l2 == nil {
        return l1
    }
    
    result := new(ListNode)
    tail := result

    for l1 != nil && l2 != nil {
        if l1.Val < l2.Val {
            tail.Next = l1
            l1 = l1.Next
        } else {
            tail.Next = l2
            l2 = l2.Next
        }
        tail = tail.Next
    }

    if l1 == nil {
        tail.Next = l2
    }

    if l2 == nil {
        tail.Next = l1
    }
    
    return result.Next
}

递归

// 执行用时：4 ms, 在所有 Go 提交中击败了67.37%的用户
// 内存消耗：2.5 MB, 在所有 Go 提交中击败了38.88%的用户
func mergeTwoLists(l1 *ListNode, l2 *ListNode) *ListNode {
    if l1 == nil {
        return l2
    }
    if l2 == nil {
        return l1
    }

    if l1.Val < l2.Val {
        l1.Next = mergeTwoLists(l1.Next, l2)
        return l1
    } else {
        l2.Next = mergeTwoLists(l2.Next, l1)
        return l2
    }
}

总结

• 习惯了Java，写go还是有点别扭
• 这题能用递归是没想到的，要多做多培养思维