LeetCode 21 合并两个有序链表

题目:

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

解题:

逐个比较,按升序排列

代码:


func mergeTwoLists(list1 *ListNode, list2 *ListNode) *ListNode {
    list := &ListNode{0, nil}
    node := list
    for list1 != nil && list2 != nil {
        if list1.Val > list2.Val {
            list.Next = &ListNode{list2.Val, nil}
            list2 = list2.Next
        } else {
            list.Next = &ListNode{list1.Val, nil}
            list1 = list1.Next
        }
        list = list.Next
    }
    if list1 == nil {
        list.Next = list2
    } else {
        list.Next = list1
    }
    return node.Next
}

官方解答:


// 递归
func mergeTwoLists(list1 *ListNode, list2 *ListNode) *ListNode {
    if list1 == nil {
        return list2
    } else if list2 == nil {
        return list1
    } else if list1.Val < list2.Val {
        list1.Next = mergeTwoLists(list1.Next, list2)
        return list1
    } else {
        list2.Next = mergeTwoLists(list1, list2.Next)
        return list2
    }
}


// 迭代
func mergeTwoLists(list1 *ListNode, list2 *ListNode) *ListNode {
    preHead := &ListNode{0, nil}
    prev := preHead
    for list1 != nil && list2 != nil {
        if list1.Val <= list2.Val {
            prev.Next = list1
            list1 = list1.Next
        } else {
            prev.Next = list2
            list2 = list2.Next
        }
        prev = prev.Next
    }
    if list1 == nil {
        prev.Next = list2
    } else {
        prev.Next = list1
    }
    return preHead.Next
}

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注