LeetCode - 兩數相加

題目描述

給出兩個 非空 的鏈表用來表示兩個非負的整數。其中，它們各自的位數是按照 逆序 的方式存儲的，並且它們的每個節點只能存儲 一位 數字。

如果，我們將這兩個數相加起來，則會返回一個新的鏈表來表示它們的和。

您可以假設除了數字 0 之外，這兩個數都不會以 0 開頭。

示例：

輸入：(2 -> 4 -> 3) + (5 -> 6 -> 4)
輸出：7 -> 0 -> 8
原因：342 + 465 = 807

我的思路

由於鏈表存儲整數是逆序存儲的，所以題目相對簡化了。只需另外開闢一個新鏈表，將兩個鏈表每一位的數值相加，將其和的個位加入到新的鏈表，當然還要計算進位，每次都要加上進位。這裏還用到了啞結點的知識，將新節點的頭節點設爲啞結點，最後返回的是頭節點的下一個節點，這樣做的好處是可以不用再判斷新節點是否爲空而導致的問題。

/**
 * Definition for singly-linked list.
 * struct ListNode {
 *     int val;
 *     ListNode *next;
 *     ListNode(int x) : val(x), next(NULL) {}
 * };
 */
class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode* res = new ListNode(0);
        ListNode* temp = res;
        int carry = 0, sum = 0;
        if(l1 == NULL) return l2;
        if(l2 == NULL) return l1;
        while(l1 != NULL || l2 != NULL){
            int x = (l1 != NULL) ? l1->val : 0;
            int y = (l2 != NULL) ? l2->val : 0;
            sum = x + y + carry;
            if(sum >= 10 && temp != NULL) {
                temp->next = new ListNode(sum % 10);
                carry = 1;
            }
            if(sum < 10 && temp != NULL){
                temp->next = new ListNode(sum);
                carry = 0;
            }
            if(l1 != NULL) l1 = l1->next;
            if(l2 != NULL) l2 = l2->next;
            temp = temp->next;
        }
        if(carry == 1) temp->next = new ListNode(1);
        return res->next;
    }
};