将两个有序链表合并为一个新的有序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例:
输入:1->2->4, 1->3->4 输出:1->1->2->3->4->4
# Definition for singly-linked list.
class ListNode:
def __init__(self, x):
self.val = x
self.next = None
class Solution:
def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
if l1 is None:
return l2
if l2 is None:
return l1
res = ListNode(None)
cur = res
while l1 is not None and l2 is not None:
if l1.val < l2.val:
cur.next = ListNode(l1.val)
cur = cur.next
l1 = l1.next
else:
cur.next = ListNode(l2.val)
cur = cur.next
l2 = l2.next
if l1 is None:
cur.next = l2
else:
cur.next = l1
return res.next
# 递归方法
def mergeTwoLists2(self, l1: ListNode, l2: ListNode) -> ListNode:
if not l1: return l2
if not l2: return l1
if l1.val < l2.val:
l1.next = self.mergeTwoLists(l1.next, l2)
return l1
else:
l2.next = self.mergeTwoLists(l1, l2.next)
return l2
if __name__=='__main__':
s = Solution()
l1 = ListNode(1)
l1.next = ListNode(2)
l1.next.next = ListNode(3)
l2 = ListNode(1)
l2.next = ListNode(3)
l2.next.next = ListNode(4)
res = s.mergeTwoLists2(l1, l2)
while res is not None:
print(res.val, end=" ")
res = res.next