題目:
?給出兩個 非空 的鏈表用來表示兩個非負的整數。其中,它們各自的位數是按照 逆序 的方式存儲的,并且它們的每個節點只能存儲 一位 數字。
?如果,我們將這兩個數相加起來,則會返回一個新的鏈表來表示它們的和。
?您可以假設除了數字 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;
}