题目
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外,这两个数都不会以 0 开头。
解法1
由于数字已经按照逆序的方式存储,那么按照正常的加法进行相加即可,但是尤其需要注意的是进位,如果两数之和超过9,就存在进位的处理。
进位的处理尤其需要关注最后一次计算的几位,很容易遗忘掉这里的细节。例如99 + 11 = 110,但是很容易忘记掉百位的1,得出错误的解为10。
代码如下:
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2)
{
ListNode* guard = new ListNode(0); // 初始化哨兵节点,让while循环中代码处理归一
ListNode* result = guard;
int carry = 0; // 进位值
while ((l1 != nullptr) || (l2 != nullptr) || (carry != 0)) {
// 获取左边加数
int left = 0;
if (l1 != nullptr) {
left = l1->val;
l1 = l1->next;
}
// 获取右边加数
int right = 0;
if (l2 != nullptr) {
right = l2->val;
l2 = l2->next;
}
guard->next = new ListNode((left + right + carry) % 10);
carry = (left + right + carry) / 10; // 更新进位值
guard = guard->next;
}
return result->next;
}
};
方法的时间复杂度为O(max(m, n)),m和n分别为l1和l2的链表的节点个数。空间复杂度为O(1)。执行结果如下:
解法2
在解法1中采用了非递归的方式,一般对于链表的问题,都可以采用递归的方式解决。在递归中需要注意递归终止条件,避免无限循环,从而导致栈空间溢出。代码如下:
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2)
{
int carry = 0;
return addTwoNumbers(l1, l2, carry);
}
private:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2, int carry)
{
if ((l1 == nullptr) && (l2 == nullptr) && (carry == 0)) { // 递归终止条件,注意需要将进位值判断条件加入进来
return nullptr;
}
int left = 0;
if (l1 != nullptr) {
left = l1->val;
l1 = l1->next;
}
int right = 0;
if (l2 != nullptr) {
right = l2->val;
l2 = l2->next;
}
ListNode* node = new ListNode((left + right + carry) % 10);
carry = (left + right + carry) / 10; // 更新进位值
node->next = addTwoNumbers(l1, l2, carry); // 递归调用处理
return node;
}
};
代码执行结果如下:
解法2与解法1相比较而言,代码更加简洁和归一,无特殊的哨兵节点处理。