刷题-力扣-面试题 02.05. 链表求和

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/sum-lists-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

给定两个用链表表示的整数，每个节点包含一个数位。

这些数位是反向存放的，也就是个位排在链表首部。

编写函数对这两个整数求和，并用链表形式返回结果。

示例：

输入：(7 -> 1 -> 6) + (5 -> 9 -> 2)，即617 + 295
输出：2 -> 1 -> 9，即912
进阶：思考一下，假设这些数位是正向存放的，又该如何解决呢?

示例：

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

题目分析

1. 根据题目描述，求两个用链表表示的整数到的和
2. 从低位向高位求和，使用carry表示进位

代码

/**
 * 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* res = new ListNode();
        ListNode* p = res;
        int carry = 0;
        while (l1 || l2) {
            if (l1 && l2) {
                int sum = carry + l1->val + l2->val;
                carry = sum / 10;
                p->next = new ListNode(sum % 10);
                l1 = l1->next;
                l2 = l2->next;
                p = p->next;
            } else if (l1) {
                if (carry == 1 && l1->val == 9) {
                    p->next = new ListNode(0);
                } else {
                    p->next = new ListNode(carry + l1->val);
                    carry = 0;
                }
                l1 = l1->next;
                p = p->next;
            } else if (l2) {
                // l2不存在
                if (carry == 1 && l2->val == 9) {
                    p->next = new ListNode(0);
                } else {
                    p->next = new ListNode(carry + l2->val);
                    carry = 0;
                }
                l2 = l2->next;
                p = p->next;
            }
        }
        if (carry) {
            p->next = new ListNode(1);
        }
        p = res->next;
        delete res;
        return p;
    }
};