程序员面试金典面试题 02.05. 链表求和

Posted 2020-12-10 galaxy-hao

题目

给定两个用链表表示的整数，每个节点包含一个数位。
这些数位是反向存放的，也就是个位排在链表首部。
编写函数对这两个整数求和，并用链表形式返回结果。

示例：

输入：(7 -> 1 -> 6) + (5 -> 9 -> 2)，即617 + 295
输出：2 -> 1 -> 9，即912
进阶：假设这些数位是正向存放的，请再做一遍。

示例：

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

思路

两个链表依次相加放入一个新链表中。

代码

时间复杂度：O(n)
空间复杂度：O(1)

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