[LeetCode] Add Two Numbers 兩個數字相加

Grandyang發表於2014-11-29
LeetCode

 

You are given two non-empty linked lists representing two non-negative integers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.

You may assume the two numbers do not contain any leading zero, except the number 0 itself.

Example:

Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8
Explanation: 342 + 465 = 807.

 

這道並不是什麼難題，演算法很簡單，連結串列的資料型別也不難，就是建立一個新連結串列，然後把輸入的兩個連結串列從頭往後擼，每兩個相加，新增一個新節點到新連結串列後面。為了避免兩個輸入連結串列同時為空，我們建立一個dummy結點，將兩個結點相加生成的新結點按順序加到dummy結點之後，由於dummy結點本身不能變，所以我們用一個指標cur來指向新連結串列的最後一個結點。好，可以開始讓兩個連結串列相加了，這道題好就好在最低位在連結串列的開頭，所以我們可以在遍歷連結串列的同時按從低到高的順序直接相加。while迴圈的條件兩個連結串列中只要有一個不為空行，由於連結串列可能為空，所以我們在取當前結點值的時候，先判斷一下，若為空則取0，否則取結點值。然後把兩個結點值相加，同時還要加上進位carry。然後更新carry，直接 sum/10 即可，然後以 sum%10 為值建立一個新結點，連到cur後面，然後cur移動到下一個結點。之後再更新兩個結點，若存在，則指向下一個位置。while迴圈退出之後，最高位的進位問題要最後特殊處理一下，若carry為1，則再建一個值為1的結點，程式碼如下：

 

C++ 解法： 

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode *dummy = new ListNode(-1), *cur = dummy;
        int carry = 0;
        while (l1 || l2) {
            int val1 = l1 ? l1->val : 0;
            int val2 = l2 ? l2->val : 0;
            int sum = val1 + val2 + carry;
            carry = sum / 10;
            cur->next = new ListNode(sum % 10);
            cur = cur->next;
            if (l1) l1 = l1->next;
            if (l2) l2 = l2->next;
        }
        if (carry) cur->next = new ListNode(1);
        return dummy->next;
    }
};

 

Java 解法：

public class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        ListNode dummy = new ListNode(-1);
        ListNode cur = dummy;
        int carry = 0;
        while (l1 != null || l2 != null) {
            int d1 = l1 == null ? 0 : l1.val;
            int d2 = l2 == null ? 0 : l2.val;
            int sum = d1 + d2 + carry;
            carry = sum >= 10 ? 1 : 0;
            cur.next = new ListNode(sum % 10);
            cur = cur.next;
            if (l1 != null) l1 = l1.next;
            if (l2 != null) l2 = l2.next;
        }
        if (carry == 1) cur.next = new ListNode(1);
        return dummy.next;
    }
}

 

在CareerCup上的這道題還有個Follow Up，把連結串列存的數字方向變了，原來是表頭存最低位，現在是表頭存最高位，請參見我的另一篇部落格2.5 Add Two Numbers 兩個數字相加 。

 

類似題目：

Multiply Strings

Add Binary

Sum of Two Integers 

Add Strings 

Add Two Numbers II  

 

參考資料：

https://leetcode.com/problems/add-two-numbers/

https://leetcode.com/problems/add-two-numbers/discuss/997/c%2B%2B-Sharing-my-11-line-c%2B%2B-solution-can-someone-make-it-even-more-concise

 

LeetCode All in One 題目講解彙總(持續更新中...)

相關文章