[LeetCode] 2. Add Two Numbers(两数相加)

Add Two Numbers

You are given two non-empty linked lists representing two non-negative integers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.

You may assume the two numbers do not contain any leading zero, except the number 0 itself.

Example:

Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8
Explanation: 342 + 465 = 807.

题目要点

• 数字按位逆序存储
• 注意进位，求和时加入上一位的进位标记值
• 链表最后一位相加出现进位时，结果链表新增节点保存进位
• 使用哑节点简化代码，函数返回链表为哑节点的下一个节点
• 注意节点移动，避免死循环

解决方案

方法一

思路

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

图1，对两数相加方法的可视化: 342 + 465 = 807，每个结点都包含一个数字，并且数字按位逆序存储。

算法

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

伪代码如下：

• 将进位 carry 初始化为 0
• 判断 l1 和 l2 是否为 NULL，计算个位值的和，初始化结果链表首节点，获取进位标记赋值 carry
• 遍历列表 l1 和 l2 直至到达它们中出现 NULL
  • 取 l1 和 l2 节点的值及进位标记 carry，累加赋值给 sum
  • 更新进位值 carry = sum / 10
  • 创建一个数值为 (sum mod 10) 的新节点，并将其设置为结果链表当前结点的下一个节点，然后将当前结点前进到下一个节点
  • 同时 l1 和 l2 前进至下一节点
• 若 l1 或 l2 其中任意节点为 NULL，对该链表单独执行上述步骤，结果链表新增节点
• 检查 carry = 1 是否成立，如果成立，则向结果链表追加一个含有数字 1 的新结点
• 返回结果链表

注意以下情况：

测试用例	说明
l1 = [0, 1] l2 = [0, 1, 2]	当一个链表比另一个链表长时。
l1 = [] l2 = [0, 1]	当一个链表为空时，即出现空列表，题目限制为非空链表，但最好增加相关判断
l1 = [9, 9] l2 = [1]	求和运算最后可能出现额外的进位，这一点很容易被遗忘

ListNode *addTwoNumbers(ListNode *l1, ListNode *l2)
{
    int carry = 0;
    int sum   = 0;
    if (l1)
    {
        sum += l1->val;
        l1 = l1->next;
    }

    if (l2)
    {
        sum += l2->val;
        l2 = l2->next;
    }
    carry = sum / 10;

    ListNode *result     = new ListNode(sum % 10);
    ListNode *tmp_result = result;
    while (l2 && l1)
    {
        sum   = l1->val + l2->val + carry;
        carry = sum / 10;
        sum %= 10;
        ListNode *tmp    = new ListNode(sum);
        tmp_result->next = tmp;
        tmp_result       = tmp_result->next;

        l2 = l2->next;
        l1 = l1->next;
    }

    while (l1)
    {
        sum              = l1->val + carry;
        carry            = sum / 10;
        ListNode *tmp    = new ListNode(sum % 10);
        tmp_result->next = tmp;
        tmp_result       = tmp_result->next;
        l1               = l1->next;
    }

    while (l2)
    {
        sum              = l2->val + carry;
        carry            = sum >= 10 ? 1 : 0;
        ListNode *tmp    = new ListNode(sum % 10);
        tmp_result->next = tmp;
        tmp_result       = tmp_result->next;
        l2               = l2->next;
    }

    if (carry > 0)
    {
        ListNode *tmp    = new ListNode(1);
        tmp_result->next = tmp;
        tmp_result       = tmp_result->next;
    }

    return result;
}

复杂度分析：

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

---

方法二（代码优化）

使用哑节点简化代码逻辑，如果没有哑结点，则必须编写额外的条件语句来初始化表头的值。

优化链表 NULL 值判断，合并重复代码，减少代码量。

public:
ListNode *addTwoNumbers(ListNode *l1, ListNode *l2)
{
    int carry = 0;

    ListNode *result     = new ListNode(0);
    ListNode *tmp_result = result;

    while (l1 || l2 || carry > 0)
    {
        int sum = 0;
        if (l1) sum += l1->val;
        if (l2) sum += l2->val;
        sum += carry;
        carry            = sum / 10;
        tmp_result->next = new ListNode(sum % 10);
        tmp_result       = tmp_result->next;
        if (l1) l1 = l1->next;
        if (l2) l2 = l2->next;
    }

    return result->next;
}

复杂度分析：

• 同上