题目
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
题目解析及优化
本题虽然是中等难度,解法比较单一,就是模拟手算加法式子,但是其中有很多容易出错的细节问题
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode* ans_head = new ListNode(0),*cur = ans_head;
int sum = 0,carry = 0;
while(l1!=NULL||l2!=NULL){
//l1,l2耗尽后用0代替
int p = l1!=NULL?l1->val:0;
int q = l2!=NULL?l2->val:0;
sum = p+q+carry;
carry = sum /10;
sum%=10;
cur->next = new ListNode(sum);
cur = cur->next;
if(l1!=NULL)l1 = l1->next;
if(l2!=NULL)l2 = l2->next;
}
//处理最后可能的进位
if(carry){
cur->next = new ListNode(carry);
}
//返回next的方法以后多应用
return ans_head->next;
}
};
再次优化一下空间:其实答案不用另外开辟空间存储,直接以其中一个作为答案即可,首先想到的是,用长的来充当答案,但是判断的时候会额外地增加代码量;而且链表的拼接特别简单,直接修改next指针即可。所以这里直接指定链表一存储答案。
- 在公共部分,正常加进位
- 当其中一个加数到头之后,继续处理另一个后面的进位
- 处理最后的进位
贴上代码:
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode *ans = l1;
//两个数字的公共位数部分
while(l1->next != nullptr && l2->next != nullptr){
l1->val += l2->val;
if (l1->val > 9){
l1->next->val += (l1->val / 10);
l1->val = l1->val % 10;
}
l1 = l1->next;
l2 = l2->next;
}
//如果b不是较短的,将b后半段接到a后面
if(l1->next == nullptr)l1->next = l2->next;
//该元素由于while循环的缘故还没有处理
l1->val += l2->val;
while(l1->next != nullptr){
if(l1->val>9){
l1->next->val += l1->val/10;
l1->val%=10;
}
l1 = l1->next;
}
//处理最后可能出现的进位问题
if(l1->val>9){
l1->next = new ListNode(l1->val/10);
l1->val%=10;
}
return ans;
}
};
其实因为边界问题处理的粗糙,导致看上面的代码时很费劲,而且写起来也极为不爽。但是没有创建新链表,时间复杂度和空间复杂度都很低。
心得及总结
- 在想要开辟空间存放答案的时候,可以先思考是否能够用已有的去存储;
- 在链表遍历的时候,用next指针不等于空的方法;
- 对于链表而言,创建新链表耗时挺长,尽量用现成的;
- 用创建链表存储答案时,可以先用虚拟头部,最后返回虚拟头部的next。