LeetCode 2.两数相加

时间:2024-03-26 07:08:35

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

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

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

直接模拟即可:

/**
 * 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 *node1 = l1;
        ListNode *node2 = l2; 
        bool carry = false;
        ListNode *ans = nullptr;
        ListNode *cur = nullptr;
        while (node1 || node2)
        {
            int val1 = node1 ? node1->val : 0;
            int val2 = node2 ? node2->val : 0;

            int ansVal = val1 + val2 + carry;
            carry = ansVal >= 10;
            ansVal %= 10;
            ListNode *ansNode = new ListNode(ansVal);
            if (ans == nullptr)
            {
                ans = ansNode;
                cur = ansNode;
            }
            else
            {
                cur->next = ansNode;
                cur = cur->next;
            }

            if (node1)
            {
                node1 = node1->next;
            }
            if (node2)
            {
                node2 = node2->next;
            }
        }

        if (carry)
        {
            cur->next = new ListNode(1);
        }

        return ans;
    }
};

如果l1的长度为n,l2的长度为m,则此算法时间复杂度为O(m+n),空间复杂度为O(1)。