Go 算法 leetcode 021 合并两个有序链表

题目

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

链接

leetocde 021 合并两个有序链表

答案

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func mergeTwoLists(list1 *ListNode, list2 *ListNode) *ListNode {
    result := &ListNode{}
    tmpRes := result

    for list1 != nil && list2 != nil {
        if list1.Val < list2.Val {
            tmpRes.Next = list1
            list1 = list1.Next
        } else if list1.Val > list2.Val {
            tmpRes.Next = list2
            list2 = list2.Next
        } else {
            tmpRes.Next = list1
            list1 = list1.Next
            tmpRes = tmpRes.Next
            tmpRes.Next = list2
            list2 = list2.Next
        }

        tmpRes = tmpRes.Next
    }

    if list1 != nil {
        tmpRes.Next = list1
    }

    if list2 != nil {
        tmpRes.Next = list2
    }

    return result.Next
}