2. 两数相加

题目

2. 两数相加

给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。

请你将两个数相加,并以相同形式返回一个表示和的链表。

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

示例 1:

输入:l1 = [2,4,3], l2 = [5,6,4]
输出:[7,0,8]
解释:342 + 465 = 807.

示例 2:

输入:l1 = [0], l2 = [0]
输出:[0]

示例 3:

输入:l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
输出:[8,9,9,9,0,0,0,1]

提示:

  • 每个链表中的节点数在范围 [1, 100] 内
  • 0 <= Node.val <= 9
  • 题目数据保证列表表示的数字不含前导零

解答

/**
 * Definition for singly-linked list.
 * public class ListNode {
 *     public var val: Int
 *     public var next: ListNode?
 *     public init() { self.val = 0; self.next = nil; }
 *     public init(_ val: Int) { self.val = val; self.next = nil; }
 *     public init(_ val: Int, _ next: ListNode?) { self.val = val; self.next = next; }
 * }
 */
class Solution {
    func addTwoNumbers(_ l1: ListNode?, _ l2: ListNode?) -> ListNode? {
        /*
        哑:哑节点开头,哑节点固定,结果返回哑节点的后继
        标:标识两个遍历的节点和一个生成的新节点
        移:while循环向前走
        进:无法移动后再处理进位
        */
        let dummyHead = ListNode()
        var currentNode = dummyHead
        var carry = 0

        var p = l1
        var q = l2

        while p != nil || q != nil {
            /*
            只要不是都走到尽头,都要进入这里
            */
            let pValue = p?.val ?? 0
            let qValue = q?.val ?? 0
            var sum = pValue + qValue + carry

            // 1、是否要进位
            carry = sum / 10
            // 2、新节点的后继(value为当前相加取余)
            currentNode.next = ListNode(sum % 10)

            // 3、往后移动
            currentNode = currentNode.next!
            p = p?.next
            q = q?.next
        }

        /*
        两个单链表都走到尽头了,如果存在进位,要追加一个后继
        这个节点的value只能是1,实际是最高位
        */
        if carry > 0 {
            currentNode.next = ListNode(carry)
        }

        return dummyHead.next
        
    }
}