題目:https://leetcode-cn.com/problems/add-two-numbers/description/
給定兩個非空鏈表來表示兩個非負整數。位數按照逆序方式存儲,它們的每個節點只存儲單個數字。將兩數相加返回一個新的鏈表。
你可以假設除了數字 0 之外,這兩個數字都不會以零開頭。
示例:
輸入:(2 -> 4 -> 3) + (5 -> 6 -> 4) 輸出:7 -> 0 -> 8 原因:342 + 465 = 807
關鍵點:進位(這里滿十加一)、指針移動(這里移向下一個節點)
class Solution: def addTwoNumbers(self, l1, l2): """ :type l1: ListNode :type l2: ListNode :rtype: ListNode """ ans = ListNode(0) #新建一個節點,初始值為0 temp = ans tempsum = 0 while True: if (l1 != None): tempsum = l1.val + tempsum #l1鏈表節點值添加到總和里 l1 = l1.next #指針指向下一個節點 if (l2 != None): tempsum = tempsum + l2.val #l2鏈表節點值添加到總和里 l2 = l2.next #指針指向下一個節點 temp.val = tempsum % 10 #取余數(滿十進位),賦值當前節點值 print(tempsum) tempsum = int(tempsum / 10) #獲取進位數賦值給總和(比如tempsum為10則進1位,否則進位為0),下一次節點相加,從新的總和開始。 if l1 == None and l2 == None and tempsum == 0: #直到沒有進位了,同時節點位空了,跳出循環。(這里加上tempsum==0條件是因為,最后兩個節 break #點和值可能大於10) temp.next = ListNode(0) #新建下一個節點,存放和 temp = temp.next #指針指向下一個節點 return ans
時間復雜度:O(max(l1長度,l2長度)) ,空間復雜度:O(max(l1長度,l2長度))