LeetCode 2. 两数相加

题目描述

解题思路

我们使用变量来跟踪进位，并从包含最低有效位的表头开始模拟逐位相加的过程。

算法

就像你在纸上计算两个数字的和那样，我们首先从最低有效位也就是列表 $l1$ 和 $l2$ 的表头开始相加。由于每位数字都应当处于 $0\dots 9$ 的范围内，我们计算两个数字的和时可能会出现 “溢出”。例如，$5+7=12$。在这种情况下，我们会将当前位的数值设置为 $2$，并将进位 $carry=1$ 带入下一次迭代。进位 $carry$ 必定是 $0$ 或 $1$，这是因为两个数字相加（考虑到进位）可能出现的最大和为 $9+9+1=19$。

伪代码如下：

1. 将当前结点初始化为返回列表的哑结点。
2. 将进位 $carry$ 初始化为 0。
3. 将 $p$ 和 $q$ 分别初始化为列表 $l1$ 和 $l2$ 的头部。
4. 遍历列表 $l1$ 和 $l2$ 直至到达它们的尾端。
5. 将 $x$ 设为结点 $p$ 的值。如果 $p$ 已经到达 $l1$ 的末尾，则将其值设置为 $0$。
6. 将 $y$ 设为结点 $q$ 的值。如果 $q$ 已经到达 $l2$ 的末尾，则将其值设置为 $0$。
7. 设定 $sum=x+y+carry$。
8. 更新进位的值，$carry=sum/10$。
9. 创建一个数值为 $(sum\mathrm{m}\mathrm{o}\mathrm{d}10)$的新结点，并将其设置为当前结点的下一个结点，然后将当前结点前进到下一个结点。
10. 同时，将 $p$ 和 $q$ 前进到下一个结点。
11. 检查 $carry=1$ 是否成立，如果成立，则向返回列表追加一个含有数字 1 的新结点。
12. 返回哑结点的下一个结点。

代码

# 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:
        re = ListNode(0)
        r = re
        carry = 0
        while (l1 or l2):
            x = l1.val if l1 else 0
            y = l2.val if l2 else 0
            s = carry + x + y
            r.next = ListNode(s % 10)
            r = r.next
            carry = s // 10
            if l1 is not None:
                l1 = l1.next
            if l2 is not None:
                l2 = l2.next
        if carry > 0:
            r.next = ListNode(1)
        return re.next

复杂度分析

• 时间复杂度：$O(\max (m,n))$，假设 $m$ 和 $n$ 分别表示 $l1$ 和 $l2$
  的长度，上面的算法最多重复 $\max (m,n)$ 次。
• 空间复杂度：$O(\max (m,n))$， 新列表的长度最多为 $\max (m,n)+1$。