LeetCode-2. 两数相加

难度:中等

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/add-two-numbers

1. 问题描述

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

请你将两个数相加,并以相同形式返回一个表示和的链表。你可以假设除了数字 0 之外,这两个数都不会以 0 开头。

2. 分析与解答

链表指针的操作,对两个链表分别设置一个指针,以实现两个位置的数的加和。但是需要注意,加和的过程中需要保留一个进位标志,用于判断是否需要进位。

/**
 * 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* p1 = l1;
        ListNode* p2 = l2;
        ListNode* head = new ListNode(0);
        ListNode* pn = head;
        int add = 0; // 进位标志
        while (p1 && p2) {
            int all = p1->val + p2->val + add;
            int cur = all % 10;
            add = all / 10;
            ListNode* p = new ListNode(cur);
            pn->next = p;
            pn = pn->next;
            p1 = p1->next;
            p2 = p2->next;
        }
        if (p1 || p2) {
            ListNode* left = (p1==nullptr)?p2:p1;
            while (left) {
                int all = left->val + add;
                int cur = all % 10;
                add = all / 10;
                ListNode* p = new ListNode(cur);
                pn->next = p;
                pn = pn->next;
                left = left->next;
            }
        }
        if (add != 0) {
            ListNode* p = new ListNode(add);
            pn->next = p;
            pn = pn->next;
        }
        return head->next;
    }
};