牛客刷题日记链表篇——链表相加（二）

当我看到这个题的时候，第一时间感觉两个数的位数是相等的，但读过题之后才知道不可行。然后就像是不是可以转成longlong类型的呢？很遗憾，不可以。之后才知道了为什么是要用链表存储数字，是因为数字太大了，以至于longlong都不可以存储，所以，链表存储数字出现了。

        顺一下解题思路吧，我们先将链表的每一个元素出到一个整形栈中，这样我们从栈顶弹出的元素的位数不会有差别，然后按位相加的结果出到另外一个整形栈中，这样新的栈是按照位数高低排好的。我们可以利用传入的两个链表元素进行返回，不需要额外的开辟空间。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */

class Solution {
  public:
    /**
     *
     * @param head1 ListNode类
     * @param head2 ListNode类
     * @return ListNode类
     */
    ListNode* addInList(ListNode* head1, ListNode* head2) {
        // write code here
        if (head1 == nullptr) return head2;
        if (head2 == nullptr) return head1;
        stack<int> stack1;//存储第一个列表元素的整形栈
        stack<int> stack2;//存储第二个列表元素的整形栈
        stack<int> ret;//返回的整形栈，可以使用上面已经定义好的，可以做代码优化
        int gap = 0, top1, top2;//定义三个整形，分别存储进位信息，第一个栈的栈顶，第二个栈顶
        ListNode* list1 = head1, *retlist = head1, *list2 = head2;
        while (head1 || head2) {//这个while循环直接将两个链表的元素分别入栈
            if (head1) {
                stack1.push(head1->val);
                head1 = head1->next;
            }
            if (head2) {
                stack2.push(head2->val);
                head2 = head2->next;
            }
        }
        while (!stack1.empty() || !stack2.empty()) {//初始化返回栈，这个要求大家可以看懂三元表达式
            top1 = stack1.empty() ? 0 : stack1.top();
            top2 = stack2.empty() ? 0 : stack2.top();
            stack1.empty() ? (void)0 : stack1.pop();
            stack2.empty() ? (void)0 : stack2.pop();
            ret.push((top1 + top2 + gap) % 10);
            gap = (top1 + top2 + gap) >= 10 ? 1 : 0;
        }
        gap == 1 ? ret.push(gap) : (void)0;//最后可能会进一位，此处二元表达式判断是否需要入栈
        while (!ret.empty()) {//构造返回的链表
            list1->val = ret.top();
            ret.pop();
            if (!ret.empty()) {
                list1 = list1->next ? list1->next : (list1->next = list2);
            }
        }
        list1->next = nullptr;
        return retlist;
    }
};