LC-02-两数相加

02 两数相加

原题链接：两数相加

个人解法

思路：
因为两个链表是倒序的，故遍历两个链表逐位相加，同时记录进位。
因为两个链表长度不一致，故仍对剩余的链表进行加法操作（此时加进位且记录进位）。
注意若最后进位大于0，则需要开一个新结点作为最终的进位。

时间复杂度：$O(n)$

代码：

/**
 * 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) {
        int k = 0;
        ListNode *p1 = l1, *p2 = l2;
        ListNode *newList = new ListNode();
        ListNode *cur = newList;
        while(p1 && p2) {
            int t = p1->val + p2->val + k;
            cur->next = new ListNode(t % 10);
            k = t / 10;
            cur = cur->next;
            p1 = p1->next;
            p2 = p2->next;
        }
        while(p1) {
            int t = p1->val + k;
            cur->next = new ListNode(t % 10);
            k = t / 10;
            cur = cur->next;
            p1 = p1->next;
        }
        while(p2) {
            int t = p2->val + k;
            cur->next = new ListNode(t % 10);
            k = t / 10;
            cur = cur->next;
            p2 = p2->next;
        }
        if(k) {
            cur->next = new ListNode(k);
        }
        return newList->next;
    }
};

更好的解法

思路：
可以将三个循环结合为一个，对于是否遍历到某个链表的尾部可以在循环内判断。

代码：

/**
 * 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) {
        auto dummy = new ListNode(-1), cur = dummy;
        int t = 0;
        while(l1 || l2 || t){
            if(l1) t += l1->val, l1 = l1->next;
            if(l2) t += l2->val, l2 = l2->next;
            cur->next = new ListNode(t % 10);
            cur = cur->next;
            t /= 10;
        }
        return dummy->next;
    }
};