图解LeetCode第 445 号问题： 两数相加 II

总第64篇/程序员小吴

LeetCode上第 445 号问题：Add Two Numbers II

题目描述

给定两个非空链表来代表两个非负整数。数字最高位位于链表开始位置。它们的每个节点只存储单个数字。将这两数相加会返回一个新的链表。

你可以假设除了数字 0 之外，这两个数字都不会以零开头。

进阶:

如果输入链表不能修改该如何处理？换句话说，你不能对列表中的节点进行翻转。

示例:

输入: (7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4)    
输出: 7 -> 8 -> 0 -> 7

思路解析

由于计算时要保证最右边的数对齐，那么很自然的想到先用 栈 存放链表中的每个值，然后依次计算。由于相加时可能产生进位，所以使用一个 flag 表示是否有进位。

提示：若栈中元素相加结束之后仍有进位，则需要新加入一个头结点。

动画演示

参考代码

 1class Solution:
 2    def addTwoNumbers(self, l1, l2):
 3        # 分别入栈
 4        stack1 = []
 5        stack2 = []
 6        while l1:
 7            stack1.append(l1.val)
 8            l1 = l1.next
 9        while l2:
10            stack2.append(l2.val)
11            l2 = l2.next
12
13        flag = 0
14        head = None
15        while stack1 or stack2 or flag != 0:
16            if stack1:
17                flag += stack1.pop()
18            if stack2:
19                flag += stack2.pop()
20            node = ListNode(flag % 10)
21            node.next = head
22            head = node
23            flag = flag // 10
24        return head

代码截图