JZ25 合并两个排序的链表 Python

# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param pHead1 ListNode类 
# @param pHead2 ListNode类 
# @return ListNode类
#
class Solution:
    def Merge(self , pHead1: ListNode, pHead2: ListNode) -> ListNode:
        # write code here
        #一个已经为空了，直接返回另一个
        if pHead1 == None:
            return pHead2
        if pHead2 == None:
            return pHead1
        #加一个表头
        head = ListNode(0)
        cur = head
        #两个链表都要不为空
        while pHead1 and pHead2:
            if pHead1.val < pHead2.val:
                cur.next = pHead1
                pHead1 = pHead1.next
                cur = cur.next#指针后移
            else:
                cur.next = pHead2
                pHead2 = pHead2.next
                cur = cur.next#指针后移
        if pHead1:
            cur.next = pHead1
        if pHead2:
            cur.next = pHead2
        return head.next