题目描述
给你两个 非空 链表来代表两个非负整数。数字最高位位于链表开始位置。它们的每个节点只存储一位数字。将这两数相加会返回一个新的链表。
你可以假设除了数字 0 之外,这两个数字都不会以零开头。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/add-two-numbers-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
逆序相加,很容易想到利用stack数据结构。利用stack先进后出的特点,首先将两个链表的数据分别压入两个栈,然后对两个栈的栈顶元素进行加法运算,并将用过的元素出栈。对于进位的处理要特别注意
class Solution {
public:
ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
stack<int> s1,s2; //声明两个栈
//入栈
while(l1){
s1.push(l1->val);
l1 = l1->next;
}
while(l2){
s2.push(l2->val);
l2 = l2->next;
}
int carry = 0; //进位值初始化为0
ListNode* dummy = nullptr;
while(!s1.empty() || !s2.empty() || carry != 0) //最高位相加时,可能产生一个进位值,并将此值入栈
{
int a = s1.empty() ? 0 : s1.top();
int b = s2.empty() ? 0 : s2.top();
int val = a + b + carry;
//出栈
if(!s1.empty()) {s1.pop();}
if(!s2.empty()) {s2.pop();}
//取进位值
carry = val/10;
//取余数
val %= 10;
//建立新结点
ListNode* cur = new ListNode(val);
cur->next = dummy;
dummy = cur;
}
return dummy;
}
};