【Leetcode】2.兩數相加

題目描述

鏈接：https://leetcode-cn.com/problems/add-two-numbers/

給出兩個 非空 的鏈表用來表示兩個非負的整數。其中，它們各自的位數是按照 逆序 的方式存儲的，並且它們的每個節點只能存儲 一位 數字。

如果，我們將這兩個數相加起來，則會返回一個新的鏈表來表示它們的和。

您可以假設除了數字 0 之外，這兩個數都不會以 0 開頭。

示例:

輸入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
輸出：7 -> 0 -> 8
原因：342 + 465 = 807

解題思路

• 將當前結點初始化爲返回列表的啞結點。
• 將進位 carry 初始化爲 00。
• 將 p 和 q 分別初始化爲列表 l1 和 l2 的頭部。
• 遍歷列表 l1 和 l2 直至到達它們的尾端。
  • 將 x 設爲結點 p 的值。如果 p 已經到達 l1 的末尾，則將其值設置爲 0。
  • 將 y 設爲結點 q 的值。如果 q 已經到達 l2 的末尾，則將其值設置爲 0。
  • 設定 sum = x + y + carry。
  • 更新進位的值，carry = sum / 10。
  • 創建一個數值爲 (sum mod 10)的新結點，並將其設置爲當前結點的下一個結點，然後將當前結點前進到下一個結點。
  • 同時，將 p 和 q 前進到下一個結點。
• 檢查 carry = 1 是否成立，如果成立，則向返回列表追加一個含有數字 1 的新結點。
• 返回啞結點的下一個結點。

代碼

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        head = ListNode(0)
        p = l1
        q = l2
        curr = head
        carry = 0

        while p or q:
            if p:
                x = p.val
            else:
                x = 0
            if q:
                y = q.val
            else:
                y = 0
            sum = carry + x + y
            carry = int(sum / 10.0)
            curr.next = ListNode(int(sum%10))
            curr = curr.next
            if p:
                p=p.next
            if q:
                q=q.next
        if carry >=1:
            curr.next = ListNode(carry)
        return head.next