本文来自https://mp.weixin.qq.com/s/IpI5ul3HI0Q0sH4EtMifvw,作者程序员吴师兄。
由于计算时要保证最右边的数对齐,那么很自然想到用 栈 存放链表中每个值,然后依次计算。由于相加时可能产生进位,所以使用一个flag表示是否有进位。
提示:如果栈中元素相加结束后仍然有进位,则需要加入一个头结点。
视频理解见链接。
public ListNode addTwoNumbers (ListNode l1, ListNode l2){
Stack<Integer> stack1 = new Stack<>();
Stack<Integer> stack2 = new Stack<>();
while (l1 != null) {
stack1.push(l1.val);
l1 = l1.next;
}
while (l2 != null) {
stack2.push(l2.val);
l2 = l2.next;
}
int flag = 0;
ListNode head = null;
while (!stack1.isEmpty() || !stack2.isEmpty() || flag!=0) {
if (!stack1.isEmpty()) {
flag += stack1.pop();
}
if (!stack2.isEmpty()) {
flag += stack2.pop();
}
ListNode node = new ListNode(flag%10); // node是一个暂时的新节点,存放当前运算出的数字,即余数
// head向前移动
node.next = head;
head = node;
flag = flag / 10; // 计算进位
}
return head;
}