每日一题-两数相加 链表问题
题目描述:
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。请你将两个数相加,并以相同形式返回一个表示和的链表。你可以假设除了数字 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]
提示:
每个链表中的节点数在范围 [1, 100] 内
0 <= Node.val <= 9
题目数据保证列表表示的数字不含前导零
来源:力扣(LeetCode)
链接:https:///problems/add-two-numbers
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;
}
};
详细解析:
当给定两个以链表形式表示的数字 l1
和 l2
时,该代码通过迭代这两个链表并进行逐位相加的方式,计算它们的和,并将结果以链表形式返回。
首先,代码定义了两个指针 head
和 tail
,并初始化为 nullptr
。这两个指针将用于构建输出链表,其中 head
指向输出链表的头部,tail
指向输出链表的尾部。
代码还定义了一个整数 carry
,初始化为 0,用于记录在相加过程中产生的进位。
接下来,代码进入一个 while
循环,只要 l1
和 l2
中还有节点,就会继续执行。
在循环中,代码首先从 l1
和 l2
中分别取出当前节点的值 n1
和 n2
,如果其中一个链表已经遍历完,则对应的值为 0。
然后,代码将 n1
、n2
和当前的进位 carry
相加,得到 sum
。这个 sum
表示当前位置上的数字相加结果。
接下来,代码根据 head
是否为空来判断是否为输出链表的第一个节点。如果 head
为空,说明当前节点是输出链表的第一个节点,将其创建为新节点,并同时设置 head
和 tail
指向该节点。
如果 head
不为空,说明输出链表已经有节点了,此时将新节点附加到 tail
后面,并更新 tail
指向新节点。
然后,代码更新进位 carry
为 sum
除以 10 的结果,以获取下一次迭代的进位。
最后,代码分别将 l1
和 l2
向后移动一位,以继续下一轮循环,直到两个链表都被遍历完。
循环结束后,如果 carry
大于 0,说明最后一次相加产生了进位,此时需要创建一个新节点表示进位,并将其附加到 tail
后面。
最终,代码返回 head
,表示输出链表的头部,即表示两个输入数字的和以链表形式返回。