1. 程式人生 > >LeetCode(2. 兩數相加)

LeetCode(2. 兩數相加)

tro leet for 進入 arr 賦值 計算 簡單 一個

問題描述

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

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

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

示例:

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

解決方案

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


class Solution:
    def addTwoNumbers(self, l1, l2):
        """
        :type l1: ListNode
        :type l2: ListNode
        :rtype: ListNode
        """
        r = res = ListNode(0)
        carry = 0
        while l1 or l2:
            x = l1.val if l1 else 0
            y = l2.val if l2 else 0
            s = carry + x + y
            carry = s // 10
            r.next = ListNode(s % 10)
            r = r.next
            if l1:
                l1 = l1.next
            if l2:
                l2 = l2.next
            if carry > 0:
                r.next = ListNode(1)
        return res.next

這個題本質上是一個簡單數學問題

  • 首先淺復制出兩個listnode,一個作為計算,一個作為最終輸出
  • carry為當前兩個節點相加以後的進位,如4+7得11那麽此時carry取值為1
  • 只要L1和L2不為空,則進入循環
  • 取L1和L2的當前值,若為空則賦值為1
  • 相加得carry
  • 隨後更新節點

LeetCode(2. 兩數相加)