数据结构与算法-7.两数相加

7、两数相加

题目

给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照逆序的方式存储的，并且每个节点只能存储一位数字。
请你将两个数相加，并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外，这两个数都不会以 0 开头。

• 每个链表中的节点数在范围 [1, 100] 内
• 0 <= Node.val <= 9
• 题目数据保证列表表示的数字不含前导零

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) {}
};

7.0、直接解法

这都没有必要解析了，很简单的。

ListNode* addTwoNumbers0(ListNode* l1, ListNode* l2) 
{
    bool flag = false;
    ListNode* pre = new ListNode(0, l1);
    ListNode* l1Node = l1;
    ListNode* l2Node = l2;
    while (l1Node || l2Node || flag)
    {
        if (!l1Node && !l2Node)             //若两个指针都为空，进位也为0，直接跳出循环
        {
            pre->next = new ListNode(1);
            break;
        }
        else if (!l1Node)                   //l1Node为空，下一个就拼接l2Node
        {
            int tempnum = l2Node->val + flag;
            flag = tempnum / 10;
            l2Node->val = tempnum % 10;
            pre->next = l2Node;
            pre = l2Node;
            l2Node = l2Node->next;
        }
        else if (!l2Node)                   //l2Node为空，下一个就拼接l1Node
        {
            int tempnum = l1Node->val + flag;
            flag = tempnum / 10;
            l1Node->val = tempnum % 10;
            pre->next = l1Node;
            pre = l1Node;
            l1Node = l1Node->next;
        }
        else                                //都不为空，下一个拼接l1Node
        {
            int tempnum = l1Node->val + l2Node->val + flag;
            l1Node->val = tempnum % 10;  
            flag = tempnum / 10;
            pre->next = l1Node;
            pre = l1Node;
            l1Node = l1Node->next;
            l2Node = l2Node->next;
        }
    }
    return l1;
}

• 时间复杂度：O（max(m, n)）m为第一条链长度，n为第二条链长度
• 空间复杂度：O（1）

7.1、优化代码版

只是做了代码的简化

ListNode* addTwoNumbers1(ListNode* l1, ListNode* l2)
{
    bool flag = false;
    ListNode* head = l1;            //这里使用l1，l2作为移动指针，head为新链表的头
    ListNode* tail = new ListNode();//tail就是上面的pre
    while (l1 || l2 || flag)
    {
        if (!l1 && !l2)             //若两个指针都为空，进位也为0，直接跳出循环
        {
            tail->next = new ListNode(1);
            return head;
        }
        int num1 = l1 == NULL ? 0 : l1->val;
        int num2 = l2 == NULL ? 0 : l2->val;
        int sumnum = num1 + num2 + flag;
        flag = sumnum / 10;
        ListNode* temp = NULL;
        if (l2)
        {
            temp = l2;
            l2 = l2->next;
        } 
        if (l1)
        {
            temp = l1;
            l1 = l1->next;
        }
        tail->next = temp;
        tail = temp;
        tail->val = sumnum % 10;
    }
    return head;;
}

• 时间复杂度：O（max(m, n)）m为第一条链长度，n为第二条链长度
• 空间复杂度：O（1）

总体来说这道题比较简单，就是需要考虑一些边界条件。