【力扣刷题C++】两数相加(链表）

非最优解，仅做记录，仅供参考

---

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

---

【题目】给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的，并且每个节点只能存储 一位 数字。请你将两个数相加，并以相同形式返回一个表示和的链表。你可以假设除了数字 0 之外，这两个数都不会以 0 开头。

• 示例 1：
  输入：l1 = [2,4,3], l2 = [5,6,4]
  输出：[7,0,8]
  解释：342 + 465 = 807.
• 示例 2：
  输入：l1 = [0], l2 = [0]
  输出：[0]
• 示例 3：
  输入：l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
  输出：[8,9,9,9,0,0,0,1]

提示：
	每个链表中的节点数在范围 [1, 100] 内
	0 <= Node.val <= 9
	题目数据保证列表表示的数字不含前导零

首次提交：很傻的思路，遍历次数偏多了，但好在至少想到了利用l1 or l2原本的空间来存放结果

执行用时：32 ms, 在所有 C++ 提交中击败了43.60%的用户
内存消耗：69.1 MB, 在所有 C++ 提交中击败了98.62%的用户

/**
 * 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) 
    {   
        // 计算两个链表长度
        int len1 = len2 = 0;
        ListNode *p1, *p2;
        for (p1 = l1; p1 != nullptr; p1 = p1->next)
            len1++;
        for (p2 = l2; p2 != nullptr; p2 = p2->next)
            len2++;
        
        // 调整长度至相同
        int temp = len1 - len2;
        if (temp > 0){
            // extend l2
            for (p2 = l2; p2 != nullptr; p2 = p2->next){
                if (p2->next == nullptr){
                    for (int i = 0; i < temp; i++){
                        p2->next = new ListNode(0);
                        p2 = p2->next;
                    }
                    break;
                }
            }
        }
        if (temp < 0){
            // extend l1
            temp = -temp;
            for (p1 = l1; p1 != nullptr; p1 = p1->next){
                if (p1->next == nullptr){
                    for (int i = 0; i < temp; i++){
                        p1->next = new ListNode(0);
                        p1 = p1->next;
                    }
                    break;
                }
            }
        }
        
        // 逐节点相加
        p1 = l1;
        for(; l1 && l2; l1 = l1->next, l2 = l2->next){
            temp = l1->val + l2->val;
            if(temp > 9){
                l1->val = temp % 10;
                if (l1->next){
                    l1->next->val += 1;
                } else{
                    l1->next = new ListNode(1);
                }
            } else{
              		l1->val = temp;
            }
        }
        return p1;
    }
};

看了眼参考答案，干净整洁啊，不过额外开了一条链表的堆空间，将其优化为原地修改l1后如下

执行用时：20 ms, 在所有 C++ 提交中击败了91.57%的用户
内存消耗：69 MB, 在所有 C++ 提交中击败了99.83%的用户

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode* p = l1;
        ListNode* prev = nullptr; // 用于记录l1的最后一个节点, p会跑到nullptr
        short andOne = 0;
		// 先在公共长度上进位求和,结果存在l1上
        while (l1 && l2) {
            int num1 = l1->val;
            int num2 = l2->val;
            int sum = num1 + num2 + andOne;
            l1->val = sum % 10;
            andOne = sum / 10;

            prev = l1;
            l1 = l1->next;
            l2 = l2->next;
        }
		// 如果是l1先到头了，后面的不用加了直接把l2放在l1末尾
        if (l2) {
            prev->next = l2; // prev存放的是l1尾结点
            l1 = prev->next;
        }
		// 如果l2先到头则表明l1后面的节点啥也不用加了
		
		// 根据之前有一方到头之前的进位情况，对l1后面部分进行处理
        while (l1 && andOne) {
            int sum = l1->val + andOne;
            l1->val = sum % 10;
            andOne = sum / 10;

            prev = l1;
            l1 = l1->next;
        }
		// l1到尾巴上了 还有进位，那就开辟新的node 存放它
        if (andOne > 0) {
            prev->next = new ListNode(andOne); 
        }

        return p;
    }
};