python實現合並兩個有序列表的示例代碼

題目描述

兩個升序鏈表合並為一個新的升序鏈表並返回。新鏈表是通過拼接給定的兩個鏈表的所有節點組成的。

LeetCode原題地址:https://leetcode-cn.com/problems/merge-two-sorted-lists/

測試用例

示例1

在這裡插入圖片描述

輸入:l1 = [1,2,4], l2 = [1,3,4]
輸出:[1,1,2,3,4,4]

示例2

輸入:l1 = [], l2 = []
輸出:[]

示例3

輸入:l1 = [], l2 = [0]
輸出:[0]

代碼詳解

因為LeetCode服務器上已經封裝瞭鏈表類,在本地測試時我需要自己來實現鏈表類,代碼如下

class ListNode:
  def __init__(self, val, next=None):
    if isinstance(val,int):
      self.val = val
      self.next = next
    elif isinstance(val,list):
      self.val = val[0]
      self.next = None
      head = self
      for i in range(1,len(val)):
        node = ListNode(val[i],None)
        head.next = node
        head = head.next

遞歸法

遞歸法的思路比較簡單,我們需要先判斷鏈表l1鏈表l2是否為空,如果為空直接返回另一個鏈表即可就不需要進行比較瞭。如果不為空,我們就需要比較鏈表節點的值誰的更大,如果l1大於l2我們就更改鏈表l2的下一個節點,然後再比較l2的下一個節點和l1,反之可得另一種情況的處理方法。

class Solution:
  def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
    #如果鏈表l1為None直接返回鏈表l2即可
    if l1 is None:
      return l2
    #如果鏈表l2為None直接返回鏈表l1即可
    elif l2 is None:
      return l1
    #如果鏈表l1大於鏈表l2
    elif l1.val > l2.val:
      #更改鏈表l2下一個節點的指向
      l2.next = self.mergeTwoLists(l1,l2.next)
      return l2
    else:
      #更改鏈表l1下一個節點的指向
      l1.next = self.mergeTwoLists(l1.next,l2)
      return l1


l1 = ListNode([1,2,4])
l2 = ListNode([1,3,4])
s = Solution()
l = s.mergeTwoLists(l1,l2)
while l:
  print(l.val)
  l = l.next

遍歷法

這個算法更簡單瞭,我們隻需要遍歷鏈表l1和l2然後再比較大小即可,對於最後沒遍歷完的部分,直接追加到合並鏈表的後面即可。

class Solution:
  def mergeTwoLists(self, l1: ListNode, l2: ListNode) -> ListNode:
    #用來合並鏈表
    prehead = ListNode(-1)
    #創建一個哨兵節點
    pre = prehead
    while l1 and l2:
      if l1.val > l2.val:
        pre.next = l2
        l2 = l2.next
      else:
        pre.next = l1
        l1 = l1.next
      #更改哨兵節點的下一個指向
      pre = pre.next
    pre.next = l1 if l1 else l2

    return prehead.next


l1 = ListNode([1,2,4])
l2 = ListNode([1,3,4])
s = Solution()
l = s.mergeTwoLists(l1,l2)
while l:
  print(l.val)
  l = l.next

參考:合並兩個有序鏈表

到此這篇關於python實現合並兩個有序列表的示例代碼的文章就介紹到這瞭,更多相關python 合並兩個有序列表內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!