給出兩個(gè) 非空 的鏈表用來(lái)表示兩個(gè)非負(fù)的整數(shù)。其中,它們各自的位數(shù)是按照 逆序 的方式存儲(chǔ)的,并且它們的每個(gè)節(jié)點(diǎn)只能存儲(chǔ) 一位 數(shù)字。
如果,我們將這兩個(gè)數(shù)相加起來(lái),則會(huì)返回一個(gè)新的鏈表來(lái)表示它們的和。
您可以假設(shè)除了數(shù)字 0 之外,這兩個(gè)數(shù)都不會(huì)以 0 開頭。
例子: 輸入:(2 -> 4 -> 3) + (5 -> 6 -> 4) 輸出:7 -> 0 -> 8
解題思路
這個(gè)思路就是我們做加法運(yùn)算, 先算個(gè)位, 如果大于10則進(jìn)1, 然后在計(jì)算十位, 以此類推.
所以遍歷兩個(gè)鏈表, 將val相加, 如果大于10, 則保留進(jìn)位1, 將結(jié)果的個(gè)數(shù)為保留, 以此類推.
class Solution {
func addTwoNumbers(_ l1: ListNode?, _ l2: ListNode?) -> ListNode? {
let result: ListNode = ListNode(0)
var p: ListNode? = l1
var q: ListNode? = l2
var head = result
// 創(chuàng)建carry為進(jìn)位
var carry = 0
while (p != nil)||(q != nil) {
//兩數(shù)相加并加上進(jìn)位
let sum = (p?.val ?? 0) + (q?.val ?? 0) + carry
//大于10則為1, 小于10則為0
carry = sum / 10
//余數(shù)則為該節(jié)點(diǎn)的值
head.next = ListNode(sum%10)
head = head.next!
p = p?.next
q = q?.next
}
if carry > 0 {
head.next = ListNode(carry)
}
return result.next
}
}
來(lái)源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/add-two-numbers