Leetcode Meet Me算法學(xué)習(xí):2.兩數(shù)相加

給出兩個(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)
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

友情鏈接更多精彩內(nèi)容