两数相加
题目描述
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例1:
输入:l1 = [2,4,3], l2 = [5,6,4]
输出:[7,0,8]
解释:342 + 465 = 807.
示例2:
输入:l1 = [0], l2 = [0]
输出:[0]
示例3:
输入:l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
输出:[8,9,9,9,0,0,0,1]
算法思想
两个链表依次从个位开始,刚好方便计算。对两个链表的值与上一位相加的进位进行相加,所得到结果的余数作为新链表结点的值,除数作为下一个待计算结点的进位。依次遍历计算两个链表得到新的链表表示俩个数相加的结果。如果两个链表的长度不同,则可以认为长度短的链表的后面有若干个 0。此外,如果链表遍历结束后,有进位不为0,还需要在答案链表的后面附加一个节点,节点的值为该进位。
代码实现
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode* head = nullptr;
ListNode* tail = nullptr;
int carry = 0;
while (l1 || l2) {
int a, b;
if (l1) {
a = l1->val;
}
else if(l1==nullptr) {
a = 0;
}
if (l2) {
b = l2->val;
}
else if (l2 == nullptr) {
b = 0;
}
int sum = a + b + 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);
tail = tail->next;
}
return head;
}