1. 程式人生 > >【LeetCode 中等題】1-兩數之和

【LeetCode 中等題】1-兩數之和

宣告:

今天是中等題第1道題。求2個連結串列之和。以下所有程式碼經過樓主驗證都能在LeetCode上執行成功,程式碼也是借鑑別人的,在文末會附上參考的部落格連結,如果侵犯了博主的相關權益,請聯絡我刪除

(手動比心ღ( ´・ᴗ・` ))

正文

題目:給出兩個 非空 的連結串列用來表示兩個非負的整數。其中,它們各自的位數是按照 逆序 的方式儲存的,並且它們的每個節點只能儲存 一位 數字。如果,我們將這兩個數相加起來,則會返回一個新的連結串列來表示它們的和。

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

示例:

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

解法1。常規解法,新開闢一條連結串列儲存兩個連結串列之和,並用carry記錄當前進位情況,程式碼如下。

執行用時: 200 ms, 在Add Two Numbers的Python3提交中擊敗了20.01% 的使用者

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def addTwoNumbers(self, l1, l2):
        """
        :type l1: ListNode
        :type l2: ListNode
        :rtype: ListNode
        """
        if not l1:
            return l2
        if not l2:
            return l1
        
        carry = 0
        dummy = ListNode(0)
        head = dummy
        while not l1 or not l2:
            sum_m = 0
            if l1:
                sum_m = l1.val
                l1 = l1.next
            if l2:
                sum_m += l2.val
                l2 = l2.next
            sum_m += carry
            carry = sum_m//10
            dummy.next = ListNode(sum_m%10)
            dummy = dummy.next
        if carry == 1:
            dummy.next = ListNode(1)
        return head.next
            
        

解法2。可以選擇就地修改l1,當其中1條更短的連結串列遍歷完了就遍歷剩下的部分,程式碼如下。

執行用時: 152 ms, 在Add Two Numbers的Python3提交中擊敗了79.06% 的使用者 

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def addTwoNumbers(self, l1, l2):
        """
        :type l1: ListNode
        :type l2: ListNode
        :rtype: ListNode
        """
        sum_m = 0
        carry = 0
        head = ListNode(0)
        head.next = l1
        while l1 and l2:
            sum_m = l1.val + l2.val + carry
            carry = sum_m//10
            l1.val = sum_m%10
            prev = l1
            l1 = l1.next
            l2 = l2.next
        l = l1 or l2    # 取較長連結串列沒被遍歷過的那一部分
        prev.next = l
        while l and carry:
            sum_m = l.val+carry
            l.val = sum_m%10
            carry = sum_m//10
            prev = l
            l = l.next
        if carry == 1:
            prev.next = ListNode(1)
        return head.next

結尾

解法1:官方解法

解法2:LeetCode