神奇補0解決連結串列相加:LeeCode002兩數相加

RiverLi發表於2019-05-07

題目描述

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

如果,我們將這兩個數相加起來,則會返回一個新的連結串列來表示它們的和。

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

示例:

輸入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
輸出:7 -> 0 -> 8
原因:342 + 465 = 807
複製程式碼

解法一:暴力計算

該解法使用數字相加的方法進行計算,但該解法有一個致命的問題是整數溢位。所以,嚴格意義上說,這種方式是錯誤的。程式碼如下:

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};

int listNodeLength(ListNode *list) {
    if (list == NULL || list->next == NULL)
    {
        return 0;
    }
    
    int size = 0;
    ListNode *tmpPoint = list;
    while(tmpPoint->next != NULL) {
        size++;
        tmpPoint = tmpPoint->next;
    }
    return size;
}


class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        int sum1 = getListSum(l1);
        int sum2 = getListSum(l2);
        int sum3 = sum1 + sum2;
        
        int sum3Length = 0;
        int tmp = sum3;
        while(tmp > 0) {
            tmp = tmp/10;
            sum3Length++;
        }
        
        ListNode *head = NULL;
        ListNode *tail = nullptr;
        int m = sum3;
        while(sum3Length > 0) {
            int val = m%(10);
            m = m / 10;
            
            ListNode *list = new ListNode(val);
            if (head)
            {
                tail->next = list;
                tail = list;
            } else {
                head = list;
                tail = head;
            }
            sum3Length--;
        }
        return head;
        
    }
    
    int getListSum(ListNode *list) {
        int length = listNodeLength(list);
        
        int sum = 0;
        int index = 0;
        ListNode *tmpPoint = list;
        while(tmpPoint != NULL) {
            sum += pow(10, length-index) * tmpPoint->val;
            tmpPoint = tmpPoint->next;
            index++;
        }
        
        return sum;
        
    }
};
複製程式碼

解法二:按位計算

該解法使用一個變數表示進位值。如下圖,temp表示進位。將兩個list長度改為一致,通過補0操作。

企業微信20190507011341.png

具體程式碼如下:

struct ListNode {
    int val;
    ListNode *next;
    ListNode(int x) : val(x), next(NULL) {}
};

int listNodeLength(ListNode *list) {
    if (list == NULL)
    {
        return 0;
    }
    
    int size = 0;
    ListNode *tmpPoint = list;
    while(tmpPoint != NULL) {
        size++;
        tmpPoint = tmpPoint->next;
    }
    return size;
}


class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        int l1Length = listNodeLength(l1);
        int l2Length = listNodeLength(l2);
        
        int max = l1Length;
        ListNode *maxListNode = l1;
        int min = l2Length;
        ListNode *minListNode = l2;
        if (l1Length<l2Length)
        {
            max = l2Length;
            maxListNode = l2;
            
            min = l1Length;
            minListNode = l1;
        }
        
        ListNode *minTail = minListNode;
        ListNode *maxTail = maxListNode;
        while (maxTail != NULL) {
            if (minTail -> next == NULL) {
                minTail->next = new ListNode(0);
            }
            maxTail = maxTail -> next;
            minTail = minTail->next;
        }
        
        int temp = 0;
        ListNode* l1temp = minListNode;
        ListNode* l2temp = maxListNode;
        ListNode *result = NULL;
        ListNode *resultTail = NULL;
        for (int i = 0; i < max; ++i)
        {
            int sum = l1temp->val + l2temp->val + temp;
            temp = sum/10;
            int val = sum%10;
            ListNode *node = new ListNode(val);
            if (result == NULL)
            {
                result = node;
                resultTail = result;
            }
            else
            {
                resultTail->next = node;
                resultTail = node;
            }
            
            l1temp = l1temp->next;
            l2temp = l2temp->next;
        }
        
        if (temp > 0) {
            resultTail->next = new ListNode(temp);
        }
        return result;
    }
};
複製程式碼

原始碼

你可以通過點選這裡獲取原始碼:LeeCodeTrain

討論

關於沒到題目的解法我都發布在我的部落格:riverli.me 和我的微信公眾號。

歡迎你關注我的公眾號,一起討論:演算法、前端、iOS、Android、使用者體驗、架構、程式設計師發展等內容。

RiverLi的公眾號

相關文章