链表-中等:两数相加 By Swift

440 阅读1分钟
  • 题目 给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。

如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。

您可以假设除了数字 0 之外,这两个数都不会以 0 开头。

示例:

输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
  • 思路

遍历两个链表相加即可,因涉及到头结点所以使用哨兵模式。

  • 代码
class Solution {
    func addTwoNumbers(_ l1: ListNode?, _ l2: ListNode?) -> ListNode? {
        var cur1 = l1
        var cur2 = l2
        let dummy: ListNode? = ListNode(0)
        var curNode = dummy
        
        var carry = 0
        // 当两个链表为空,且无进位时(carry  == 0),终止循环
        while cur1 != nil || cur2 != nil || carry != 0 {
            var value1 = 0
            var value2 = 0
            var sum = 0
            
            if let node = cur1 {
                value1 = node.val
                cur1 = cur1?.next
            }
            
            if let node = cur2 {
                value2 = node.val
                cur2 = cur2?.next
            }
            sum = value1 + value2 + carry
            carry = sum / 10
            curNode?.next = ListNode(sum % 10)
            curNode = curNode?.next
        }
        return dummy?.next
    }
}