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;
}
复杂度分析:
- 同上