leetcode 經典面試題之連結串列 (2. 兩數相加)
阿新 • • 發佈:2019-01-02
- 兩數相加
給出兩個 非空 的連結串列用來表示兩個非負的整數。其中,它們各自的位數是按照 逆序 的方式儲存的,並且它們的每個節點只能儲存 單位 數字。
如果,我們將這兩個數起來相加起來,則會返回出一個新的連結串列來表示它們的和。
您可以假設除了數字 0 之外,這兩個數都不會以 0 開頭。
示例:
輸入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
輸出:7 -> 0 -> 8
原因:342 + 465 = 807
func addTwoNumbers(l1 *ListNode, l2 *ListNode) * ListNode {
res := new(ListNode)
cur := res
carry := false
for l1 != nil || l2 != nil {
sum := 0
if l1 != nil {
sum = sum + l1.Val
l1 = l1.Next
}
if l2 != nil {
sum = sum + l2.Val
l2 = l2.Next
}
if carry {
sum++
}
if sum > 9 {
sum = sum - 10
carry = true
} else {
carry = false
}
cur.Next = &ListNode{Val: sum}
cur = cur.Next
}
if carry {
cur.Next = &ListNode{Val: 1}
}
return res.Next
}