力扣2. 两数相加

Problem: 2. 两数相加

题目描述

思路

1.创建虚拟头节点dummy，用于存储相加的结果数字；
2.让指针p1、p2、tail分别指向l1、l2、dummy，定义int变量carry记录每次相加的进位值；
3.当p1不为空或者p2不为空时，定义int变量sum：

3.1 若p1不为空：sum += p1 -> val;
3.2 若p2不为空：sum += p2 -> val;
3.3 若craay不为空: sum += carry;
3.4 生成当前的结果（tail -> next = new ListNode(sum % 10);），并更新carry(carry = sum / 10;)
3.5 最后遍历完l1与l2后carry还不为0，则再将其添加到结果链表末尾

复杂度

时间复杂度:

$O(max(m,n))$;其中$m$为链表l1的长度，$n$为链表l2的长度

空间复杂度:

$O(m+n)$

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:
    /**
     *
     * @param l1 Given linked list l1
     * @param l2 Given linked list l2
     * @return ListNode*
     */
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode* p1 = l1;
        ListNode* p2 = l2;
        int carry = 0;
        ListNode* dummy = new ListNode(666);
        ListNode* tail = dummy;
        if (l1 == nullptr) {
            return l2;
        }
        if (l2 == nullptr) {
            return l1;
        }
        while (p1 != nullptr || p2 != nullptr) {
            int sum = 0;

            if (p1 != nullptr) {
                sum += p1 -> val;
                p1 = p1 -> next;
            }
            if (p2 != nullptr) {
                sum += p2 -> val;
                p2 = p2 -> next;
            }
            if (carry != 0) {
                sum += carry;
            }
            tail -> next = new ListNode(sum % 10);
            carry = sum / 10;
            tail = tail -> next;
        }
        //If carry is not equal to 0
        if (carry != 0) {
            tail -> next = new ListNode(carry);
        }
        return dummy -> next;
    }
};