題目:
將兩個升序鏈表合並為一個新的 升序 鏈表並返回。新鏈表是通過拼接給定的兩個鏈表的所有節點組成的。
示例:
輸入:1->2->4, 1->3->4
輸出:1->1->2->3->4->4
解題思路:
直接使用遞歸,也就是整體法進行編寫,假設其中第一個鏈表的數字比另一個數小,則將后面的兩個鏈表使用merge函數打亂重新排序然后得到最后的結果。代碼如下:
# Definition for singly-linked list. # class ListNode: # def __init__(self, val=0, next=None): # self.val = val # self.next = next class Solution: def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
#假設l1當中啥都沒有,則返回l2,因為l2當中的node一定是升序排列的
if l1==None: return l2 if l2==None: 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
方法2
這道題除了使用遞歸之外,還可以使用迭代的方法求解,使用迭代的方法求解也非常簡單。這里可以采用merge sort(歸並排序)當中進行merge(歸並)的一個思想。 也就是說我們可以分別在兩個鏈表的表頭對鏈表的value進行比較,將較小的一個node放到新的,我們需要進行返回的鏈表當中。然后鏈表的指針向后移動一格,再進行新一輪的比較。如果兩個鏈表的表頭的大小是相同的話,則將其均放入到新的需要返回的鏈表當中。
代碼如下:
# Definition for singly-linked list. # class ListNode: # def __init__(self, val=0, next=None): # self.val = val # self.next = next class Solution: def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode: ret=ListNode(0) real_ret=ret while l1 or l2: if l1!=None and l2!=None: one=l1.val two=l2.val if one<two: ret.next=ListNode(one) ret=ret.next l1=l1.next if two<one: ret.next=ListNode(two) ret=ret.next l2=l2.next if one==two: ret.next=ListNode(two) ret=ret.next ret.next=ListNode(two) ret=ret.next l1=l1.next l2=l2.next if l1!=None and l2==None: ret.next=ListNode(l1.val) ret=ret.next l1=l1.next if l1==None and l2!=None: ret.next=ListNode(l2.val) ret=ret.next l2=l2.next return real_ret.next
