Leetcode-2： 两数相加

题目链接

https://leetcode-cn.com/problems/add-two-numbers/

题目

给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的，并且每个节点只能存储 一位 数字。

请你将两个数相加，并以相同形式返回一个表示和的链表。

你可以假设除了数字 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
• 题目数据保证列表表示的数字不含前导零

思路

遍历两个链表，逐一进行模拟加法进位，如果遍历到了最后，两个指针都指向了空而还有进位，则需要再补上一个结点，具体见代码注释。

C++ Code

/**
 * 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) {
        ListNode* head1=l1; //head1遍历链表1
        ListNode* head2=l2; //head2遍历链表2
        ListNode* pre1=head1; //pre1记录head1的前一个结点 便于在遍历完还有进位时再接一个结点
        ListNode* pre2=head2; //pre2记录head2的前一个结点
        //先把结果全部放到l1上
        int flag=0; //进位符
        //当两个链表都还没遍历完
        while(head1!=nullptr&&head2!=nullptr)
        {
            if(head1->val+head2->val+flag>=10)
            {
                head1->val= head1->val + head2->val+ flag - 10;
                flag=1;
            }
            else
            {
                head1->val= head1->val + head2->val+ flag;
                flag=0;
            }
            pre1=head1; pre2=head2;
            head1=head1->next; head2=head2->next;
        }
        //1. 当链表1已空而链表2非空 继续计算链表2 链表1尾部接上链表2当前位置
        if(head1==nullptr&&head2!=nullptr) 
        {
            //链表1尾部接上链表2
            pre1->next=head2;
            while(head2!=nullptr)
            {
                if(head2->val+flag>=10)
                {
                    head2->val= 0;
                    flag=1;
                }
                else
                {
                    head2->val= head2->val+ flag;
                    flag=0;
                    //falg=0意味着后面的数字不会再发生改变了 此时可以返回结果
                    return l1;
                }
                pre2=head2;
                head2=head2->next;
            }
            //全部遍历完了还有余数，说明是1+99999这种情况 需要增加一位
            if(flag==1)
            {
                ListNode* end=new ListNode(1,nullptr);
                pre2->next=end;
                return l1;
            }
        }

        //2. 当链表1非空而链表2已空 和上面类似
        else if(head1!=nullptr&&head2==nullptr) 
        {
            while(head1!=nullptr)
            {
                if(head1->val+flag>=10)
                {
                    head1->val= 0;
                    flag=1;
                }
                else
                {
                    head1->val= head1->val+ flag;
                    flag=0;
                    //后面的数字不会再发生改变了 可以返回结果
                    return l1;
                }
                pre1=head1;
                head1=head1->next;
            }
            //遍历完了还有余数，说明是1+99999这种情况 需要增加一位
            if(flag==1)
            {
                ListNode* end=new ListNode(1,nullptr);
                pre1->next=end;
                return l1;
            }

        }

        //3.当链表1和链表2都已空 判断是否还剩进位
        else if(flag==1)
        {
            ListNode* end=new ListNode(1,nullptr);
            pre1->next=end;
            return l1;
        }
        return l1;

    }
};

结果