21_合并两个有序链表

"""
将两个有序链表合并为一个新的有序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。 
示例：
输入：1->2->4, 1->3->4
输出：1->1->2->3->4->4
"""


from ListNode import ListNode
from PrintNode import PrintList

# 方法一 迭代
def mergeTwoLists(node1, node2):
    head = ListNode(0)
    temp = head

    while node1 and node2:
        if node1.val <= node2.val:
            head.next = node1
            node1 = node1.next
        else:
            head.next = node2
            node2 = node2.next
        head = head.next
    if node1:
        head.next = node1
    elif node2:
        head.next = node2

    return temp.next


# 方法二 递归
def mergeTwoLists(node1, node2):
    if node1 is None:
        return node2
    elif node2 is None:
        return node1

    if node1.val < node2.val:
        node1.next = mergeTwoLists(node1.next, node2)
        return node1
    else:
        node2.next = mergeTwoLists(node1, node2.next)
        return node2


node1 = ListNode(1)
node2 = ListNode(3)
node3 = ListNode(5)
node4 = ListNode(2)
node5 = ListNode(4)
node6 = ListNode(6)
node7 = ListNode(7)

node1.next = node2
node2.next = node3

node4.next = node5
node5.next = node6
node6.next = node7
PrintList(mergeTwoLists(node1, node4))