兩數相加
題目
給出兩個 非空 的連結串列用來表示兩個非負的整數。其中,它們各自的位數是按照逆序的方式儲存的,並且它們的每個節點只能儲存一位數字。
如果,我們將這兩個數相加起來,則會返回一個新的連結串列來表示它們的和。
您可以假設除了數字 0 之外,這兩個數都不會以 0 開頭。
示例:
輸入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
輸出:7 -> 0 -> 8
原因:342 + 465 = 807
複製程式碼
Go程式設計實現
思路:迭代計算每位數字的和,記錄進位。當全部數位和進位計算完畢,結果得到。
程式碼如下:
// 單向連結串列結構
type ListNode struct {
Val int
Next *ListNode
}
func AddTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
//判斷是否存在運算元為 0 的特例
if l1.Val == 0 && l1.Next == nil {
return l2
}
if l2.Val == 0 && l2.Next == nil {
return l1
}
result := &ListNode{}
//迭代處理的結果節點,從第一個開始
node := result
// 進位
var carry int
// 和
v := 0
for {
// 若 l1 還有,則累加,同時切換下一個l1
if l1 != nil {
v += l1.Val
l1 = l1.Next
}
// 若 l2 還有,則累加,同時切換下一個 l2
if l2 != nil {
v += l2.Val
l2 = l2.Next
}
//累加進位
v += carry
carry = v / 10 // 進位
node.Val = v % 10 // 當前位
// 若 l1 l2 進位都沒了,結束
if l1 == nil && l2 == nil && carry == 0 {
break
}
// 構建下一個結果節點
node.Next = &ListNode{}
node = node.Next
v = 0
}
return result
}
複製程式碼
本實現在 LeetCode 上測試通過,結果為:
執行用時 : 8 ms , 在所有 golang 提交中擊敗了 96.84% 的使用者。
記憶體消耗 : 4.9 MB , 在所有 golang 提交中擊敗了 94.01% 的使用者。
關注 紅牛慕課,或進入 紅牛慕課 獲取更多程式設計題。