Leetcode第二題:兩數相加

題目:

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

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

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

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/add-two-numbers
著作權歸領釦網絡所有。商業轉載請聯繫官方授權,非商業轉載請註明出處。

個人思路:

兩鏈表同時遍歷,有進位就爲下一個進位的值+1。注意特殊情況!

官方答案推薦:

同上。若不是逆序,則可用棧,依舊要從最小位開始計算。

python代碼:

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

class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        curNode = ListNode(0)
        l3 = curNode
        carry = 0
        while l1 or l2:
            val1 = l1.val if l1 else 0
            val2 = l2.val if l2 else 0

            curVal = val1 + val2 + carry
            if(curVal>=10):
                curVal = curVal % 10
                carry = 1
            else:
                carry = 0
            
            curNode.next = ListNode(curVal)
            curNode = curNode.next
            if l1:
                l1 = l1.next
            if l2:
                l2 = l2.next
        if(carry>0):
            curNode.next = ListNode(1)
        return l3.next

反思:

①寫代碼的時候一直糾結第一個節點的值如何再循環中保存,start = curNode,之後curNode指向鏈表最後,start指向val = 0的頭節點,那麼返回的時候返回start.next就可以了。

②代碼冗餘,不用判斷node.next == None,可以直接判斷if node來判斷

③最後一步進位一定要注意

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章