Python 中用連結串列表示數字的程式


假設我們有兩個單鏈表 L1 和 L2,每個連結串列都表示一個數字,最低有效位在前,我們需要找到它們的和表示的連結串列。

因此,如果輸入類似於 L1 = [5,6,4] L2 = [2,4,8],則輸出將是 [7, 0, 3, 1, ]

為了解決這個問題,我們將遵循以下步驟

  • 進位 := 0

  • 結果 := 一個值為 0 的新節點

  • 當前 := 結果

  • 當 L1 不為空或 L2 不為空或進位不為零時,執行以下操作

    • l0_val := L1 的值,如果 L1 不為空,否則為 0

    • l1_val := L2 的值,如果 L2 不為空,否則為 0

    • 總和 := l0_val + l1_val

    • 進位 := (總和 + 進位) / 10 的商

    • 加值 := (總和 + 進位) / 10 的餘數

    • 當前.next := 一個值為加值的新節點

    • 當前 := 當前的下一個節點

    • L1 := L1 的下一個節點,如果 L1 不為空,否則為 null

    • L2 := L2 的下一個節點,如果 L2 不為空,否則為 null

  • 當前的下一個節點 := null

  • 返回結果的下一個節點

讓我們看看下面的實現以更好地理解

示例

class ListNode:
   def __init__(self, data, next = None):
      self.val = data
      self.next = next

def make_list(elements):
   head = ListNode(elements[0])
   for element in elements[1:]:
      ptr = head
      while ptr.next:
         ptr = ptr.next
      ptr.next = ListNode(element)
     
   return head

def print_list(head):
   ptr = head
   print('[', end = "")
   while ptr:
      print(ptr.val, end = ", ")
      ptr = ptr.next
   print(']')

class Solution:
   def solve(self, L1, L2):
      carry = 0
      res = ListNode(0)
      curr = res

      while L1 or L2 or carry:
         l0_val = L1.val if L1 else 0
         l1_val = L2.val if L2 else 0
         sum_ = l0_val + l1_val
         carry, add_val = divmod(sum_ + carry, 10)

         curr.next = ListNode(add_val)
         curr = curr.next

         L1 = L1.next if L1 else None
         L2 = L2.next if L2 else None

      curr.next = None

return res.next

ob = Solution()
L1 = make_list([5,6,4])
L2 = make_list([2,4,8])
print_list(ob.solve(L1, L2))

輸入

[5,6,4], [2,4,8]

輸出

[7, 0, 3, 1, ]

更新於: 2020-11-09

100 次檢視

開啟你的 職業生涯

透過完成課程獲得認證

開始學習
廣告