面试题 02.05. 链表求和
题目
面试题 02.05. 链表求和
给定两个用链表表示的整数,每个节点包含一个数位。
这些数位是反向存放的,也就是个位排在链表首部。
编写函数对这两个整数求和,并用链表形式返回结果。
示例:
输入:(7 -> 1 -> 6) + (5 -> 9 -> 2),即617 + 295
输出:2 -> 1 -> 9,即912
进阶:思考一下,假设这些数位是正向存放的,又该如何解决呢?
示例:
输入:(6 -> 1 -> 7) + (2 -> 9 -> 5),即617 + 295
输出:9 -> 1 -> 2,即912
解答
var addTwoNumbers = function(l1, l2) {
let dummy = new ListNode(0), cur = dummy;
let overflow = false;
while (l1 && l2) {
let val = l1.val + l2.val + (overflow ? 1 : 0);
if (val >= 10) {
overflow = true;
val = val - 10;
}
else {
overflow = false;
}
cur.next = new ListNode(val);
cur = cur.next;
l1 = l1.next;
l2 = l2.next;
}
l1 = l2 ? l2 : l1;
while (l1) {
let val = l1.val + (overflow ? 1 : 0);
if (val >= 10) {
overflow = true;
val = val - 10;
}
else {
overflow = false;
}
cur.next = new ListNode(val);
cur = cur.next;
l1 = l1.next;
}
if (overflow) {
cur.next = new ListNode(1);
}
return dummy.next;
};