Python兩數相加

兩數相加

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

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

您可以假設除了數字 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: ListNode, l2: ListNode) -> ListNode:
        lst1 = []
        while l1:
            lst1.append(l1.val)
            l1 = l1.next
        lst2 = []
        while l2:
            lst2.append(l2.val)
            l2 = l2.next
        m, n = len(lst1), len(lst2)
        if m < n:
            m, n, lst1, lst2 = n, m, lst2, lst1
        lst3 = []
        for i in range(0, m):
            if i < n:
                num = lst1[i] + lst2[i]
            else:
                num = lst1[i]
            if num > 9:
                num = num - 10
                lst3.append(num)
                if i == m - 1:
                    lst3.append(1)
                else:
                    lst1[i + 1] += 1
            else:
                lst3.append(num)
        node = ListNode(lst3[0])
        cur = node
        for i in lst3[1:]:
            node_ = ListNode(i)
            cur.next = node_
            cur = node_
        return node

思路二:

# 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:
        jin_wei = 0
        head = cur = None
        while l1 or l2:
            l1_val = l1.val if l1 else 0
            l2_val = l2.val if l2 else 0
            l1 = l1.next if l1 else None
            l2 = l2.next if l2 else  None
            num = l1_val + l2_val
            if jin_wei == 1:
                num += 1
                jin_wei = 0
            if num > 9:
                num = num - 10
                jin_wei = 1
            if not head:
                head = ListNode(num)
                cur = head
            else:
                node = ListNode(num)
                cur.next = node
                cur = node
        if jin_wei == 1:
            cur.next = ListNode(1)
        return head