Swift3.0刷LeetCote笔记：Add Two Numbers

题目

您将获得两个非空链接列表，表示两个非负整数。 数字以相反的顺序存储，并且它们的每个节点包含单个数字。 添加两个数字并将其作为链接列表返回。

您可以假定这两个数字不包含任何前导零，除了数字0本身。

例子

Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8

答案

这是leetcode上一位大佬给出的答案，我将其改成了swift语言实现。我的思路是遍历两个链表，得两个数字，相加，但是最后求出返回链表时出了问题。

思路很简单，我们从链表的第一位开始加起，并定义一个变量来记录进位（如果俩数相加为10位，那么这个变量carry为1，否则carry为0）。

就像你在一张纸上计算两个数字的和一样，我们从最低有效数字开始，这个最低有效数字就是链表 l1 和 l2 的表头。由于每个数位都在 0~ 9 的范围内，所以对两个数字求和可能会“溢出”，比如 5 + 7 = 12 。在这种情况下，我们将当前数字设为2，并将 carry = 1 带到下一次计算。carry只能是0 或 1 ，因为两个数字的最大可能的和是9 + 9 + 1 = 19 。

伪代码

1. 将当前节点初始化为返回列表的虚拟表头

2. 初始化carry变量为0 （记录“溢出”情况）

3. 分别将 p 和 q 初始化为 l1 和 l2 的头

4. 遍历链表 l1 和 l2，直到到达两端

 4.1 将节点 p 的值赋值给x，如果 p 已经到达 l1 的末尾，设置为0
 4.2 将节点 q 的值赋值给y，如果 q 已经到达 l2 的末尾，设置为0
 4.3 设置和 sum = x + y + carry
 4.4 更新 carry = sum / 10
 4.5 创建一个值为 sum 余 10 (sum%10) 的新节点，并将其设置为当前节点的 next ，然后将当前节点推进到下一个节点
 4.6 推进节点 p 和 q

5 当结束循环后，判断carry的值。如果carry值为1，则表明还要进位1位，创建一个新的结点并赋值为1

6 返回虚拟表头的next结点 

代码

public class ListNode {
    public var val: Int
    public var next: ListNode?
    public init(_ val: Int) {
        self.val = val
        self.next = nil
    }
}

func addTwoNumbers(_ l1: ListNode?, _ l2: ListNode?) -> ListNode? {
    let pHead: ListNode? = ListNode(0)
    var p: ListNode? = l1!
    var q: ListNode? = l2!
    var curr: ListNode! = pHead

    var carry: Int = 0

    while p != nil || q != nil {
        let x: Int = (p != nil) ? p!.val : 0
        let y: Int = (q != nil) ? q!.val : 0
        let sum: Int = carry + x + y
        carry = sum / 10
        curr.next = ListNode(sum % 10)
        curr = curr.next
        if (p != nil) {
            p = p!.next
        }
        if q != nil {
            q = q!.next
        }
    }

    if carry > 0 {
        curr.next = ListNode(carry)
    }

    return pHead?.next
}

var l1 = ListNode(2)
l1.next = ListNode(4)
l1.next?.next = ListNode(3)

var l2 = ListNode(5)
l2.next = ListNode(6)
l2.next?.next = ListNode(4)

var l3 = addTwoNumbers(l1, l2)
print("\(l3?.val)")