审题
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
这道题,我依稀记得是大二的时候刚开始刷,刷完第一题信心满满,刷到第二题就弃坑了哈哈,不过最近因为在刷这个链表系列,于是把这道题抽出来看看,感觉要考虑的特殊情况稍微多了一点,不过逻辑还是很清晰的,于是写下如下代码。
代码实现
方案一(上文提到的方法):
码量有点吓人,不过是因为大量重复代码没封装的缘故,还有就是经验不太够,下面贴个官方题解来分析一下他比我们的代码牛在哪里。
/**
* Definition for singly-linked list.
* struct ListNode {
* int val;
* ListNode *next;
* ListNode() : val(0), next(nullptr) {}
* ListNode(int x) : val(x), next(nullptr) {}
* ListNode(int x, ListNode *next) : val(x), next(next) {}
* };
*/
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode* cur = new ListNode;
ListNode* res = new ListNode;
res = cur;
while ( l1 != NULL || l2 != NULL ) {
ListNode* tmp_Node = new ListNode;
int tmp_int;
if ( l1 == NULL ) {
cur->next = l2;
cur = cur->next;
while ( cur->val == 10 ) {
if ( cur->next != NULL ) {
cur->val = 0;
cur->next->val++;
cur = cur->next;
} else {
cur->val = 0;
cur->next = new ListNode(1);
cur = cur->next;
}
}
return res->next;
}
if ( l2 == NULL ) {
cur->next = l1;
cur = cur->next;
while ( cur->val == 10 ) {
if ( cur->next != NULL ) {
cur->val = 0;
cur->next->val++;
cur = cur->next;
} else {
cur->val = 0;
cur->next = new ListNode(1);
cur = cur->next;
}
}
return res->next;
}
tmp_int = l1->val + l2->val;
if ( l1->next != NULL ) {
if ( tmp_int >= 10 ) {
tmp_Node->val = tmp_int % 10;
l1->next->val++;
} else {
tmp_Node->val = tmp_int;
}
cur->next = tmp_Node;
cur = cur->next;
} else if ( l2->next != NULL ) {
if ( tmp_int >= 10 ) {
tmp_Node->val = tmp_int % 10;
l2->next->val++;
} else {
tmp_Node->val = tmp_int;
}
cur->next = tmp_Node;
cur = cur->next;
} else {
if ( tmp_int >= 10 ) {
tmp_Node->val = tmp_int % 10;
cur->next = tmp_Node;
cur = cur->next;
ListNode* final_Node = new ListNode;
final_Node->val = 1;
final_Node->next = NULL;
cur->next = final_Node;
cur = cur->next;
} else {
tmp_Node->val = tmp_int;
cur->next = tmp_Node;
cur = cur->next;
}
}
l1 = l1->next;
l2 = l2->next;
}
return res->next;
}
};
方案2:
官方题解如下:
首先最聪明的一点就是把不等长链表中较短的那个链表当作后面全是0,并且利用三目运算符使得程序更加清晰。
另外,将sum的整数部分保存在一个变量中可以使代码更加清晰,这个是真的妙啊,还需要再仔细品味一下。
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode *head = nullptr, *tail = nullptr;
int carry = 0;
while (l1 || l2) {
int n1 = l1 ? l1->val: 0;
int n2 = l2 ? l2->val: 0;
int sum = n1 + n2 + carry;
if (!head) {
head = tail = new ListNode(sum % 10);
} else {
tail->next = new ListNode(sum % 10);
tail = tail->next;
}
carry = sum / 10;
if (l1) {
l1 = l1->next;
}
if (l2) {
l2 = l2->next;
}
}
if (carry > 0) {
tail->next = new ListNode(carry);
}
return head;
}
};
反思
同样是链表的基本操作,但是官方题解给的就是比较简洁,而这带来的优势也是不容易出错,这是技巧,更是经验,希望能够通过反复练习掌握其精髓。