问题描述
给你两个 非空 链表来代表两个非负整数。数字最高位位于链表开始位置。它们的每个节点只存储一位数字。将这两数相加会返回一个新的链表。
你可以假设除了数字 0 之外,这两个数字都不会以零开头。
进阶:
如果输入链表不能修改该如何处理?换句话说,你不能对列表中的节点进行翻转。
示例:
输入:(7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 8 -> 0 -> 7
解题思路
不修改原链表,就需要借助额外的数据结构了,关键在于如何对末尾对其。可以使用栈来实现。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 | class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
s1, s2 = [], []
while l1:
s1.append(l1.val)
l1 = l1.next
while l2:
s2.append(l2.val)
l2 = l2.next
head, prev = None, 0
while s1 or s2 or prev:
a = s1.pop() if s1 else 0
b = s2.pop() if s2 else 0
cur = a + b + prev
prev, cur = divmod(cur, 10)
p = ListNode(cur)
p.next = head
head = p
return head
|
直接利用 Python 的大数运算。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17 | class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
a = b = 0
while l1:
a = a * 10 + l1.val
l1 = l1.next
while l2:
b = b * 10 + l2.val
l2 = l2.next
c = a + b
head = None if c else ListNode(0)
while c:
p = ListNode(c % 10)
p.next = head
head = p
c //= 10
return head
|