BM11 链表相加(二)
给你两个 非空 链表来代表两个非负整数。数字最高位位于链表开始位置。它们的每个节点只存储一位数字。将这两数相加会返回一个新的链表。
你可以假设除了数字 0 之外,这两个数字都不会以零开头。
进阶:
如果输入链表不能修改该如何处理?换句话说,你不能对列表中的节点进行翻转。
示例:
输入:(7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 8 -> 0 -> 7
解题思路:
栈
本题的主要难点在于链表中数位的顺序与我们做加法的顺序是相反的,为了逆序处理所有数位,我们可以使用栈:把所有数字压入栈中,再依次取出相加。计算过程中需要注意进位的情况。
Python代码:
# 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:
n1, n2 = [], []
while l1:
n1.append(l1.val)
l1 = l1.next
while l2:
n2.append(l2.val)
l2 = l2.next
res = None
carry = 0
while n1 or n2 or carry != 0:
if n1:
a = n1.pop()
else:
a = 0
if n2:
b = n2.pop()
else:
b = 0
t = a + b + carry
carry = t // 10
c = t % 10
cur = ListNode(c)
cur.next = res
res = cur
return res