LeetCode2之兩數相加
最近又把LeetCode練習給撿起來了,前面真的是空了差不多有一個月沒碰了,惰性作怪啊!這次準備好好練習,堅持每天一題,為了防止自己偷懶,所以就把每天練習的題目發到知乎裡督促自己吧,當然我的答案效率一般不是很高,所以如果有大神們看到了希望能多多指教!!!
2018.10.24
今天我做的是第二題:兩數相加
題目如下:給定兩個非空連結串列來表示兩個非負整數。位數按照逆序方式儲存,它們的每個節點只儲存單個數字。將兩數相加返回一個新的連結串列。
示例:
輸入:(2 -> 4 -> 3) + (5 -> 6 -> 4) 輸出:7 -> 0 -> 8 原因:342 + 465 = 807
而官方所給的解釋有:
就像你在紙上計算兩個數字的和那樣,我們首先從最低有效位也就是列表 l1l1和 l2l2 的表頭開始相加。由於每位數字都應當處於 0 \ldots 90…9 的範圍內,我們計算兩個數字的和時可能會出現“溢位”。例如,5 + 7 = 125+7=12。在這種情況下,我們會將當前位的數值設定為 22,並將進位 carry = 1carry=1 帶入下一次迭代。進位 carrycarry 必定是 00 或 11,這是因為兩個數字相加(考慮到進位)可能出現的最大和為 9 + 9 + 1 = 199+9+1=19。
虛擬碼如下:
- 將當前結點初始化為返回列表的啞結點。
- 將進位 carrycarry 初始化為 00。
- 將 pp 和 qq 分別初始化為列表 l1l1 和 l2l2 的頭部。
- 遍歷列表 l1l1 和 l2l2 直至到達它們的尾端。
- 將 xx 設為結點 pp 的值。如果 pp 已經到達 l1l1 的末尾,則將其值設定為 00。
- 將 yy 設為結點 qq 的值。如果 qq 已經到達 l2l2 的末尾,則將其值設定為 00。
- 設定 sum = x + y + carrysum=x+y+carry。
- 更新進位的值,carry = sum / 10carry
- 建立一個數值為 (sum \bmod 10)(summod10) 的新結點,並將其設定為當前結點的下一個結點,然後將當前結點前進到下一個結點。
- 同時,將 pp 和 qq 前進到下一個結點。
- 檢查 carry = 1carry=1 是否成立,如果成立,則向返回列表追加一個含有數字 11 的新結點。
- 返回啞結點的下一個結點。
請注意,我們使用啞結點來簡化程式碼。如果沒有啞結點,則必須編寫額外的條件語句來初始化表頭的值。
所給的Java程式碼如下:
public ListNode addTwoNumbers(ListNode l1, ListNode l2) {
ListNode dummyHead = new ListNode(0);
ListNode p = l1, q = l2, curr = dummyHead;
int carry = 0;
while (p != null || q != null) {
int x = (p != null) ? p.val : 0;
int y = (q != null) ? q.val : 0;
int sum = carry + x + y;
carry = sum / 10;
curr.next = new ListNode(sum % 10);
curr = curr.next;
if (p != null) p = p.next;
if (q != null) q = q.next;
}
if (carry > 0) {
curr.next = new ListNode(carry);
}
return dummyHead.next;
}
根據這些提示我寫的Python程式碼如下:
# 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
"""
head = ListNode(0)
p = l1
q = l2
curHead = head
carry = 0
while p!=None or q!=None:
if p!=None:
pVal = p.val
else:
pVal = 0
if q!=None:
qVal = q.val
else:
qVal = 0
sum = pVal + qVal + carry
carry = int(sum/10)
curHead.next = ListNode(sum%10)
curHead = curHead.next
if p!=None:
p = p.next
if q!=None:
q = q.next
if carry == 1:
curHead.next = ListNode(1)
return head.next
if __name__ == '__main__':
a = [2, 4, 3]
b = [5, 6, 4]
l1 = ListNode(0)
curL1 = l1
l2 = ListNode(0)
curL2 = l2
for i in range(len(a)):
curL1.next = ListNode(a[i])
curL1 = curL1.next
for j in range(len(b)):
curL2.next = ListNode(b[j])
curL2 = curL2.next
solution = Solution()
result = solution.addTwoNumbers(l1.next, l2.next)
l3 = []
while result != None:
l3.append(result.val)
result = result.next
print(l3)
這裡有一點是提醒大家:
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
就是這個定義節點ListNode類在解答頁面是註釋掉的,我們不用把它的註釋給去掉了,因為它內部其實已經是把它加上去了,如果你去掉了註釋,系統會報如下的錯誤
Line 65: Exception: Type <class '__main__.ListNode'>: Not implemented
最後貼出我的演算法執行效率
我有一點需要說明:就是這個執行時間有點不科學,我每次執行的時間都差的很多,有時候還差有時候很好。但總體上我覺得我目前這個演算法還是比較囉嗦的,暫時沒有找到更好地方法,後面看看能不能想出更好地方法,當然也希望各位大神們多多指教啦!