如题
将两个有序链表合并为一个新的有序链表并返回。新链表是通过拼接给定的两个链表的所有节点组成的。
示例:
输入: 1-> 2-> 4,1 -> 3-> 4
输出: 1-> 1-> 2-> 3-> 4-> 4
思路 :
1. 若都为空则直接返回空
2.若一个为空另一个不为,则直接返回另一个非空
3.非递归:建立一个新头结点和一个移动指针,通过比较大小指针指来指去,在一次迭代后,就落脚于刚刚链过的那一条。
如果中途中一个为空,则直接链接到另一条上。
代码如下:
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution(object):
def mergeTwoLists(self, l1, l2):
"""
:type l1: ListNode
:type l2: ListNode
:rtype: ListNode
"""
ptr1 = l1
ptr2 = l2
head = ListNode(0)
tail =head
while ptr1 and ptr2:
if ptr1.val < ptr2.val:
tail.next = ptr1
tail = ptr1
ptr1 = ptr1.next
tail.next = None
else:
tail.next = ptr2
tail =ptr2
ptr2 = ptr2.next
tail.next = None
if ptr1:
tail.next = ptr1
elif ptr2:
tail.next = ptr2
return head.next
递归:当然就不需要迭代啦,通过返回值来进行操作就好。
每次判断之后,.next 直接为函数名的返回值
代码如下:
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution(object):
def mergeTwoLists(self, l1, l2):
"""
:type l1: ListNode
:type l2: ListNode
:rtype: ListNode
"""
if not l1:
return l2
if not l2:
return l1
head = ListNode(None)
if l1.val <= l2.val:
head = l1
head.next = self.mergeTwoLists(l1.next,l2)
else:
head = l2
head.next = self.mergeTwoLists(l1,l2.next)
return head