給出兩個(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 開(kāi)頭。
示例:
輸入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
輸出:7 -> 0 -> 8
原因:342 + 465 = 807
public class ListNode {
public var val: Int
public var next: ListNode?
public init(_ val: Int) {
self.val = val
self.next = nil
}
func log() -> Void {
print("\(self.val)")
}
}
func getNode(_ nums: [Int]) -> ListNode? {
var firstNode: ListNode?
var beforeNode: ListNode?
for num in nums {
let node = ListNode(num)
if beforeNode != nil {
beforeNode?.next = node
}else {
firstNode = node
}
beforeNode = node
}
return firstNode
}
func addTwoNumbers(_ l1: ListNode?, _ l2: ListNode?) -> ListNode? {
var node1 = l1
var node2 = l2
let node = ListNode(-1)
var currentNode = node
//是否大于10的標(biāo)記
var carry = 0
while node1 != nil || node2 != nil {
let n1 = (node1 == nil) ? 0 : (node1?.val ?? 0)
let n2 = (node2 == nil) ? 0 : (node2?.val ?? 0)
let sum = n1 + n2 + carry
carry = (sum >= 10) ? 1 : 0
currentNode.next = ListNode(sum % 10)
currentNode = currentNode.next!
if node1 != nil {
node1 = node1?.next
}
if node2 != nil {
node2 = node2?.next
}
}
//如果最后一位計(jì)算完成后 carry == 1 (計(jì)算結(jié)果大于10) 則再添加一個(gè)節(jié)點(diǎn)
if carry == 1 {
currentNode.next = ListNode(1)
}
return node.next
}
func showLog(_ node: ListNode?) -> Void {
if node != nil {
node?.log()
if node?.next != nil {
showLog(node?.next)
}
}
}
let node1 = getNode([2, 4, 3])
let node2 = getNode([5, 6, 4])
let node = addTwoNumbers(node1, node2)
showLog(node)