LeetCode題解[Add Two Numbers II]

本文首發於個人blog中.

今天的題目是Add Two Numbers II:

題目描述：

You are given two non-empty linked lists representing two non-negative integers. The most significant digit comes first 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.

Follow up:
What if you cannot modify the input lists? In other words, reversing the lists is not allowed.

Example:

Input: (7 -> 2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 8 -> 0 -> 7

求解思路：

這道題最簡練的方法就是用棧了作了，而後還能夠先計算出兩個鏈表的長度，再生成鏈表。

思路都比較簡單，emmm，一開始覺得作成遞歸的會比較簡單，後來發現不只時間複雜度仍是代碼的複雜度都挺高的，作成迭代的會好不少。

/**
 * 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) {
        
        stack<int> s1, s2;
        int carry = 0, v1, v2, v;
        ListNode *cur;
        ListNode *pre = nullptr;
        
        while(l1 || l2) {
            if (l1) { s1.push(l1->val); l1 = l1->next; }
            if (l2) { s2.push(l2->val); l2 = l2->next; }
        }
        
       
        while(!s1.empty() || !s2.empty() || carry) {
            v1 = v2 = 0;
            if (!s1.empty()) { v1 = s1.top(); s1.pop(); }
            if (!s2.empty()) { v2 = s2.top(); s2.pop(); }
            
            v = v1 + v2 + carry;
            cur  = new ListNode(v % 10);
            cur->next = pre;
            pre = cur;
            carry = v /10;
        }      
        return pre;
    }
};

給出一個dicuss中的方法，很巧妙，用的是翻轉鏈表（題目說不能修改原來的鏈表，因此這裏是翻轉output):

ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
    int n1 = 0, n2 = 0, carry = 0;
    ListNode *curr1 = l1, *curr2 = l2, *res = NULL;
    while( curr1 ){ curr1=curr1->next; n1++; }
    while( curr2 ){ curr2=curr2->next; n2++; } 
    curr1 = l1; curr2 = l2;
    while( n1 > 0 && n2 > 0){
        int sum = 0;
        if( n1 >= n2 ){ sum += curr1->val; curr1=curr1->next; n1--;}
        if( n2 > n1 ){ sum += curr2->val; curr2=curr2->next; n2--;}
        res = addToFront( sum, res );
    }
    curr1 = res; res = NULL;
    while( curr1 ){
        curr1->val += carry; carry = curr1->val/10;
        res = addToFront( curr1->val%10, res );
        curr2 = curr1; 
        curr1 = curr1->next;
        delete curr2;
    }
    if( carry ) res = addToFront( 1, res );
    return res;
}

ListNode* addToFront( int val, ListNode* head ){
    ListNode* temp = new ListNode(val);
    temp->next = head;
    return temp;
}