给定两个非空链表来代表两个非负整数。数字最高位位于链表开始位置。它们的每个节点只存储单个数字。将这两数相加会返回一个新的链表。
你可以假设除了数字 0 之外,这两个数字都不会以零开头。
进阶:
如果输入链表不能修改该如何处理?换句话说,你不能对列表中的节点进行翻转。
示例:
输入: (7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4)
输出: 7 -> 8 -> 0 -> 7
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/add-two-numbers-ii
测试用例先行考虑:
- 输入:null + null
输出:null - 输入:null + (1 -> 2)
输出:1 -> 2 - 输入:(7 -> 2) + (4 -> 2) (最高位进位)
输出:1 -> 1 -> 4 - 输入:(7 -> 2 -> 4 -> 3) +( 5 -> 6 -> 4) (一般用例)
输出:7 -> 8 -> 0 -> 7
分析:
模仿加法思路,从低位开始,而链表为单向,所以可以用栈结构存储这些数值,每次将栈顶元素相加,有进位则保存,直到两个栈为空及没有进位为止。
/**
* Definition for singly-linked list.
* public class ListNode {
* int val;
* ListNode next;
* ListNode(int x) { val = x; }
* }
*/
class Solution {
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
Stack<Integer> stack_a = new Stack<>();
Stack<Integer> stack_b = new Stack<>();
while(l1 != null) {
stack_a.push(l1.val);
l1 = l1.next;
}
while(l2 != null) {
stack_b.push(l2.val);
l2 = l2.next;
}
ListNode head = new ListNode(-1);
int carry = 0;
while(!stack_a.empty() || !stack_b.empty() || carry==1) {
int x = stack_a.empty() ? 0 : stack_a.pop();
int y = stack_b.empty() ? 0 : stack_b.pop();
int sum = x + y + carry;
if(sum >= 10) {
sum = sum - 10;
carry = 1;
} else {
carry = 0;
}
ListNode node = new ListNode(sum);
node.next = head.next;
head.next = node;
}
return head.next;
}
}