2. Add Two Numbers(兩數相加)
阿新 • • 發佈:2019-02-02
給定兩個非空連結串列來代表兩個非負數,位數按照逆序方式儲存,它們的每個節點只儲存單個數字。將這兩數相加會返回一個新的連結串列。
你可以假設除了數字 0 之外,這兩個數字都不會以零開頭。
示例:
輸入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
輸出:7 -> 0 -> 8
原因:342 + 465 = 807
class Solution{
/**
* 002-Add Two Numbers (單鏈表表示的兩個數相加)
* @param l1 第一個數
* @param l2 第二個數
* @return 結果
*/
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
if (l1 == null) {
return l2;
}
if (l2 == null) {
return l1;
}
ListNode p1 = l1;
ListNode p2 = l2;
ListNode root = new ListNode(0); // 頭結點
ListNode r = root;
root.next = l1;
int carry = 0 ; // 初始進位
int sum;
while (p1 != null && p2 != null) {
sum = p1.val + p2.val + carry;
p1.val = sum % 10; // 本位的結果
carry = sum / 10; // 本次進位
r.next = p1;
r = r.next; // 指向最後一個相加的結點
p1 = p1.next;
p2 = p2.next;
}
if (p1 == null) {
r.next = p2;
} else {
r.next = p1;
}
// 最後一次相加還有進位
if (carry == 1) {
// 開始時r.next是第一個要相加的結點
while (r.next != null) {
sum = r.next.val + carry;
r.next.val = sum % 10;
carry = sum / 10;
r = r.next;
}
// 都加完了還有進位,就要建立一個新的結點
if (carry == 1) {
r.next = new ListNode(1);
}
}
return root.next;
}
}