题目:
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
所需类简单实现:
public class ListNode {
int val;
ListNode next;
ListNode(int x) {
val = x;
}
}
我的思路:
由于数是倒序存的,所以前面的数就是低位的数,所以通过设计一个进位标记,然后遍历然后依次累加即可。由于还得判断next
所以代码显得比较冗长。
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
boolean is = false; // 标记是否进位
ListNode result = null, r2 = null; // result是结果,r2是最后一个next
for (;;) {
int iresult = l1.val + l2.val;
if (is) {
iresult += 1; //加上进位
}
if (iresult >= 10) { //判断是否向高位进位
is = true;
iresult = iresult % 10;
} else {
is = false;
}
//存入结果
if (result == null) {
result = new ListNode(iresult);
r2 = result;
} else {
r2.next = new ListNode(iresult);
r2 = r2.next;
}
if (l1.next == null && l2.next == null) { //结束条件
break;
} else {
l1 = l1.next == null ? new ListNode(0):l1.next;
l2 = l2.next == null ? new ListNode(0):l2.next;
}
}
if (is) {
r2.next = new ListNode(1);
}
return result;
}
官方题解:
- 解题思路与我是一致的,不过代码上做了合并优化,特别是while循环里面的处理应该好好琢磨琢磨。特别注意的是这里返回的是
dummyHead.next
也蛮巧妙的,省去了链表开头或结尾的情况判断;以及carry
进位标记的设置。
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
ListNode dummyHead = new ListNode(0);
ListNode p = l1, q = l2, curr = dummyHead;
int carry = 0;
while (p != null || q != null) {
int x = (p != null) ? p.val : 0;
int y = (q != null) ? q.val : 0;
int sum = carry + x + y;
carry = sum / 10;
curr.next = new ListNode(sum % 10);
curr = curr.next;
if (p != null) p = p.next;
if (q != null) q = q.next;
}
if (carry > 0) {
curr.next = new ListNode(carry);
}
return dummyHead.next;
}