算法与数据结构(两数相加)
题目
思路
因为两个链表都是逆序存储数字的,所以我们可以通过直接将各位的数字相加求得结果即可。
我们可以用head以及tail保存相加后链表的头和尾。
(1)若l1链表不为空,则val1 = l1->val,否则为0。
(2)若l2链表不为空,则val2 = l2->val,否则为0。
sum等于两个数字和进位相加。
首先判断其第一位数字是否已经相加,若没有,则给head,tail创建一个值为sum%10的ListNode新节点。
否则,将创建一个ListNode新节点与tail相连。
若链表不为空,则继续遍历。
最后,如果所有的数都加完了且carry不为0,则在最后补上值为carry的节点。
代码
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
ListNode* head = nullptr, *tail = nullptr;
int carry = 0;
while(l1 || l2)
{
int val1 = l1 ? l1->val : 0;
int val2 = l2 ? l2->val : 0;
int sum = carry + val1 + val2;
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;
}
};