天天看点

LeetCode-两数相加题目解法

LeetCode-两数相加

  • 题目
  • 解法

题目

给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。

如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。

您可以假设除了数字 0 之外,这两个数都不会以 0 开头。

示例:

输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)

输出:7 -> 0 -> 8

原因:342 + 465 = 807

解法

就像在纸上计算两个数字的和那样,我们首先从最低有效位也就是列表 l1 和 l2的表头开始相加。由于每位数字都应当处于0···9的范围内,我们计算两个数字的和时可能会出现“溢出”。例如:5 + 6 = 11。在这种情况下,我们会将当前位的数字设置为 1,并将进位carry = 1 带入下一次迭代。进位carry必定是 0 或 1,这是因为两个数字相加(考虑到进位)可能出现的最大和为 9 + 9 + 1 = 19。

  • 将当前结点初始化为返回列表的哑结点。
  • 将进位 carry 初始化为0。
  • 将 p 和 q 分别初始化为列表 l1 和 l2 的头部。
  • 遍历列表 l1 和 l2 直至到达它们的尾端。

    (1)将x设为结点p的值。如果p已经到达l1的末尾,则将其值设置为0。

    (2)将y设为结点q的值。如果q已经到达l2的末尾,则将其值设置为0。

    (3)设定 sum = x + y + carry。

    (4)更新进位的值,carry = sum / 10。

    (5)创建一个数值为(sum mod 10)的新结点,并将其设置为当前结点的下一个结点,然后将当前结点前进到下一个结点。

    (6)同时,将p和q前进到下一个结点。

  • 检查carry = 1 是否成立,如果成立,则向返回列表追加一个含有数字 1 的新结点。
  • 返回哑结点的下一个结点。

C++代码:

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode dummy(-1);   //头结点
        int carry = 0;  //进位
        ListNode *prev = &dummy;  //链表
        for(ListNode *pa = l1, *pb = l2;
            pa != NULL || pb != NULL;
            pa = pa == NULL ? NULL : pa->next,
            pb = pb == NULL ? NULL : pb->next,
            prev = prev->next) {
            const int ai = pa == NULL ? 0 : pa->val;
            const int bi = pb == NULL ? 0 : pb->val;
            const int value = (ai + bi + carry) % 10;
            carry = (ai + bi + carry) / 10;
            prev->next = new ListNode(value);   //尾插法
        }
        if(carry > 0)
            prev->next = new ListNode(carry);
        return dummy.next;
    }
};
           

时间复杂度O(m+n),空间复杂度O(1)。

Java代码:

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) { val = x; }
 * }
 */
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
    ListNode dummyHead = new ListNode(0);
    ListNode p = l1, q = l2, curr = dummyHead;
    int carry = 0;
    while (p != null || q != null) {
        int x = (p != null) ? p.val : 0;
        int y = (q != null) ? q.val : 0;
        int sum = carry + x + y;
        carry = sum / 10;
        curr.next = new ListNode(sum % 10);
        curr = curr.next;
        if (p != null) p = p.next;
        if (q != null) q = q.next;
    }
    if (carry > 0) {
        curr.next = new ListNode(carry);
    }
    return dummyHead.next;
}
           

时间复杂度O(max(m, n)),假设m和n分别表示l1和l2的长度,上面算法最多重复max(m, n)次;

空间复杂度O(max(m, n)),新列表的长度最多为max(m, n) + 1。

如果链表中的数字不是按逆序存储,可以先考虑将其反转,然后按此法解。