目录

算法与数据结构两数相加

算法与数据结构(两数相加)

题目

https://i-blog.csdnimg.cn/direct/c482f04ba39044e0bb1053ab43105202.png

思路

因为两个链表都是逆序存储数字的,所以我们可以通过直接将各位的数字相加求得结果即可。

我们可以用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;
    }
};