1. 程式人生 > >【LeetCode】Add Two Numbers 解題報告

【LeetCode】Add Two Numbers 解題報告

【題目】

You are given two linked lists representing two non-negative numbers. 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.

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

【說明】

這道題不難,思路很直觀,考基本功。

遍歷兩個連結串列時必須的,需要注意的是最後一個結點如果有進位,需要new一個結點。

網上很多程式碼是重新new一個連結串列來存放結果,我覺得可以直接用 l1 來存放結果。

網上很多程式碼看起來更短,但時間複雜度並沒有降低,為便於理解,我覺著還是下面寫的比較清晰。

【Java程式碼】

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     int val;
 *     ListNode next;
 *     ListNode(int x) {
 *         val = x;
 *         next = null;
 *     }
 * }
 */
public class Solution {
    public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
        if (l1 == null) return l2;
        if (l2 == null) return l1;
        
        //把相加後的結果存放於連結串列1,pre1是用於最後有進位時在其後new新結點
        ListNode ret = l1;
        ListNode pre1 = new ListNode(0);
        pre1.next = l1;
        
        int flag = 0;
        while (l1 != null && l2 != null) {
            l1.val = l1.val + l2.val + flag;
            flag = l1.val / 10;
            l1.val = l1.val % 10;
            pre1 = l1;
            l1 = l1.next;
            l2 = l2.next;
        }
        
        //如果連結串列2有剩餘,接到連結串列1的後面
        if (l2 != null) {
            pre1.next = l2;
            l1 = l2;
        }
        
        while (l1 != null) {
            l1.val += flag;
            flag = l1.val / 10;
            l1.val = l1.val % 10;
            pre1 = l1;
            l1 = l1.next;
        }
        
        if (flag > 0) {
            ListNode node = new ListNode(1);
            pre1.next = node;
        }
        
        return ret;
    }
}