1. 程式人生 > >LeetCode 演算法學習(1)

LeetCode 演算法學習(1)

題目描述

Add Two Numbers
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.

題目大意

求兩個非空連結串列所代表的整數的和,並將其儲存到一個連結串列中。

思路分析

這是一道比較簡單的單鏈表題目,直接通過兩個連結串列的同時遍歷,數字相加,得到相應的進位(carry)值,再加到高一位的運算中,類似於序列的加法器原理。時間複雜度為O(Max(len1,len2)),空間複雜度也為O(Max(len1,len2))。

關鍵程式碼

class Solution {
public:
    ListNode* addTwoNumbers(ListNode* l1, ListNode* l2) {
        ListNode *itr1 = l1, *itr2 = l2;
        ListNode *result = new ListNode(-1);
        ListNode *itrResult = result;
        int carry = 0;
        while (itr1 != NULL && itr2 != NULL) { // 同時遍歷
            int tempResult = itr1->val + itr2->val + carry;
            ListNode *tempNode = new ListNode(tempResult%10);
            itrResult->next = tempNode;
            itrResult = itrResult->next;
            carry = tempResult/10;
            itr1 = itr1->next;
            itr2 = itr2->next;
        }
        // 當兩表長度不一樣時
        while (itr1 != NULL) {
            int tempResult = itr1->val + carry;
            ListNode *tempNode = new ListNode(tempResult%10);
            itrResult->next = tempNode;
            itrResult = itrResult->next;
            carry = tempResult/10;
            itr1 = itr1->next;
        }
        while (itr2 != NULL) {
            int tempResult = itr2->val + carry;
            ListNode *tempNode = new ListNode(tempResult%10);
            itrResult->next = tempNode;
            itrResult = itrResult->next;
            carry = tempResult/10;
            itr2 = itr2->next;
        }
        // 最高位有進位
        if (carry != 0) {
            ListNode *tempNode = new ListNode(carry%10);
            itrResult->next = tempNode;
        }
        return result->next;
    }
};

總結

這道題雖然比較簡單,但是在操作過程中和看完答案後,發現自己有幾個問題:

1.程式碼過於冗餘,表的長度不一樣時其實可以合併到同一個迴圈內,即可以保證易讀性,也可以提高一定的效率。
2. 在處理連結串列時不太熟練,不能快速地使用連結串列,沒有考慮帶頭指標的連結串列。
3. 考慮到除法和取餘數的操作可能相對會費時一點,這裡應該也可以用減法提高一定的效率。