C++编程练习题: 对两单向链表求和

学编程

这题难度中等, 原因就是在实现的时候细节很多, 不容易一次性搞对. 还好这两个列表表示的整数是反过来的, 所以实现上可以边遍历边相加, 只需要在累加每一位的时候记得记住进位即可.

C++单向列表定义如下:

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};

当然, 还有更笨的方法, 就是先把每个列表转换成数字, 然后想加再把结果换成列表.

我们需要创建一个dummy结点, 记住, 在函数返回的时候返回 dummy.next (下一个节点)即可. 算法复杂度是 O(A+B) 其中A和B分别是两个输入列表的长度.

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

英文: Coding Exercise – Add Two Numbers on Singly-Linked List