Swift3.0刷LeetCote笔记:Add Two Numbers

来源:互联网 发布:一般淘宝客服上班时间 编辑:程序博客网 时间:2024/05/19 12:36

题目

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

您可以假定这两个数字不包含任何前导零,除了数字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 和 q5 当结束循环后,判断carry的值。如果carry值为1,则表明还要进位1位,创建一个新的结点并赋值为16 返回虚拟表头的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)")
0 0
原创粉丝点击