02.05 链表求和

原题目

面试题 02.05. 链表求和

给定两个用链表表示的整数，每个节点包含一个数位。

这些数位是反向存放的，也就是个位排在链表首部。

编写函数对这两个整数求和，并用链表形式返回结果。

示例：

输入：(7 -> 1 -> 6) + (5 -> 9 -> 2)，即617 + 295
输出：2 -> 1 -> 9，即912

**进阶：**思考一下，假设这些数位是正向存放的，又该如何解决呢?

示例：

输入：(6 -> 1 -> 7) + (2 -> 9 -> 5)，即617 + 295
输出：9 -> 1 -> 2，即912

第一遍解法

暴力法，使用了很多个判断条件，去依次相加。

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        int flag = 0;
        
        ListNode *p1 = new ListNode(0);
        ListNode *p2 = new ListNode(0);

        p1->next = l1;
        p2->next = l2;

        while (p1->next != NULL || p2->next != NULL) {
            if (p1->next != NULL && p2->next != NULL) {
                int tmp = (p1->next->val + p2->next->val + flag) % 10;
                flag = (p1->next->val + p2->next->val + flag) / 10;
                p1->next->val = tmp;
                p1 = p1->next;
                p2 = p2->next;
            }
            else if (p1->next != NULL && p2->next == NULL) {
                int tmp = (p1->next->val + flag) % 10;
                flag = (p1->next->val + flag) / 10;
                p1->next->val = tmp;
                p1 = p1->next;
            }
            else if (p1->next == NULL && p2->next != NULL){
                int tmp = (p2->next->val + flag) % 10;
                flag = (p2->next->val + flag) / 10;
                p2->next->val = tmp;
                p1->next = new ListNode(p2->next->val);
                p1 = p1->next;
                p2 = p2->next;
            }
        }

        if (flag) {
            p1->next = new ListNode(flag);
        }

        return l1;
    }
};

时间复杂度: O(n)

空间复杂度: O(n)

• 这种方法使得判断条件太多，导致代码非常冗余。
• 空间复杂度应该可以进一步优化为O(1)

网上好的解法

思考一下发现手工求和步骤就是：

• 先对应位求和（位数少的数对应位不存在就用0加）
• 加上上一次的进位
• 得到当前位
• 记录进位
• 当位数大的数遍历完（遍历完较长的链表）且进位也为0的时候就可以停止了

class Solution
{
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2)
    {
        ListNode *head = new ListNode(-1), *p1 = l1, *p2 = l2, *p = head;//用带头节点的可以少一点初始的特判
        int sum = 0, carr = 0;
        while (p1 || p2 || carr) //如果改用&&则while结束还要多一些特判
        {
            sum = 0;//当前两位数字和
            if(p1)
            {
                sum += (p1->val);
                p1 = p1->next;
            }
            if(p2)
            {
                sum += (p2->val);
                p2 = p2->next;
            }
            sum += carr; //加上上一位的进位
            ListNode *t = new ListNode(sum % 10); //得到当前位数字
            carr = sum / 10; //得到当前位对下一位的进位
            p->next = t;//当前位连接上去
            p = p->next;//游标指针更新
        }
        return head->next;
    }
};


作者：wu-yu-61
链接：https://leetcode-cn.com/problems/sum-lists-lcci/solution/clian-biao-mo-ni-shou-gong-qiu-he-jian-dan-yi-dong/
来源：力扣（LeetCode）

时间复杂度: O(n)

空间复杂度: O(n)

最后的代码

与网上的代码差不多。

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        int sum = 0, flag = 0;
        ListNode *head = new ListNode(-1), *p = head;
        ListNode *p1 = l1, *p2 = l2;
        while (p1 || p2 || flag) {
            sum = 0;
            if (p1) {
                sum += p1->val;
                p1 = p1->next;
            }
            if (p2) {
                sum += p2->val;
                p2 = p2->next;
            }
            p->next = new ListNode((sum + flag) % 10);
            p = p->next;
            flag = (sum + flag) / 10;
            
        }
        return head->next;
    }
};

时间复杂度: O(n)

空间复杂度: O(n)

小结

• 无