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, ]
廣告