题目:
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
思路:
最初的想法:
-
链表的起点相当于个位,后面是十位、百位、千位;
所以只要逐位相加,记录进位的值
然后在每次循环中将 L1 L2 flag相加的值作为新的结点就ok了~ -
伪代码:
while(L1 && L2){ sum = flag + val1 +val2; L1、L2后移——》next; 创建新节点添加sum到链表末尾; } 处理剩余的链表,如果flag == 1表示存在进位 判断哪个链表不为空: 继续处理这个链表的值+flag,一直到最后
-
这里存在问题:
剩余的链表处理还要重复上面判断,移动指针的过程,能不能弄到一起呢????
优化的思想:
- 伪代码:
当L1 、L2 、flag只要有一个存在(flag!=0 || L1 || L2不为空)就循环: 在循环中依次判断这三个变量,符合条件的将其加入sum: sum+= ?; 移动对应的指针; 判断当前sum 是否>= 10: 如果符合就处理flag = 1、 如果否:就flag = 0; 创建一个结点存储当前的sum 并添加到返回链表的末尾;
优化代码:
/**
* 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) {
int flag = 0;
ListNode* head = new ListNode(-1);
ListNode* p = head;
while(l1 != NULL || l2 != NULL || flag)
{
int temp_sum = flag;
flag = 0;
if(l1) {
temp_sum+= l1->val;
l1 = l1->next;
}
if(l2) {
temp_sum+= l2->val;
l2 = l2->next;
}
if(temp_sum > 9) {
flag = 1;
temp_sum %= 10;
}
ListNode* Node = new ListNode(temp_sum);
p->next = Node;
p = p->next;
}
p = head->next;
delete head;
return p;
}
};