李洪强iOS经典面试题34-求两个链表表示的数的和
李洪强iOS经典面试题34-求两个链表表示的数的和
问题
给你两个链表，分别表示两个非负的整数。每个链表的节点表示一个整数位。
为了方便计算，整数的低位在链表头，例如：123 在链表中的表示方式是：
3 -> 2 -> 1现在给你两个这样结构的链表，请输出它们求和之后的结果。例如：
输入: (2 -> 4 -> 1) + (5 -> 6 -> 1)
输出: 7 -> 0 -> 3
代码模版
本题的 Swift 代码模版如下：
private class ListNode {
public var val: Int
public var next: ListNode?
public init(_ val: Int) {
self.val = val
self.next = nil
}
}
class Solution {
func addTwoNumbers(_ l1: ListNode?, _ l2: ListNode?)
-> ListNode? {
}
}考查点
本题出自 LeetCode 上的第 2 题。
这是我高中学习编程时最早接触的一类题目，我们把这类题目叫做「高精度计算」，其实就是在计算机计算精度不够时，模拟我们在纸上演算的方式来计算答案，然后获得足够精度的解。
我还记得我 7 年前第一次去网易有道面试的时候，就考查的是一道类似的高精度计算题目，比这道题复杂得多，我当时用了一个比较笨的办法，加上当时还是用 C++ 写的，内存分配和释放写起来也比较麻烦，最后写了两页 A4 纸才写完。
这道题其实完全不考查什么「算法」，人人都知道怎么计算，但是它考察了「将想法转换成代码」的能力，新手通常犯的毛病就是：意思都明白，但是写不出来代码。所以，这类题目用来过滤菜鸟确实是挺有效的办法。
答案
本题的做法其实没什么特别，就是直接计算。计算的时候需要考虑到以下这些情况：
-
两个整数长度不一致的情况。
-
进位的情况。当进位产生时，我们需要保存一个标志位，以便在计算下一位的和的时候，加上进位。
-
当计算完后，如果还有进位，需要处理最后结果加一位的情况。
以下是完整的代码，我使用了一些 Swift 语言的特性，比如用 flatMap 来减少对于 Optional类型值为 nil 的判断。
private class ListNode {
public var val: Int
public var next: ListNode?
public init(_ val: Int) {
self.val = val
self.next = nil
}
}
private class Solution {
private func getNodeValue(_ node: ListNode?) -> Int {
return node.flatMap { $0.val } ?? 0
}
func addTwoNumbers(_ l1: ListNode?, _ l2: ListNode?)
-> ListNode? {
if l1 == nil || l2 == nil {
return l1 ?? l2
}
var p1 = l1
var p2 = l2
let result: ListNode? = ListNode(0)
var current = result
var extra = 0
while p1 != nil || p2 != nil || extra != 0 {
var tot = getNodeValue(p1) +
getNodeValue(p2) + extra
extra = tot / 10
tot = tot % 10
let sum:ListNode? = ListNode(tot)
current!.next = sum
current = sum
p1 = p1.flatMap { $0.next }
p2 = p2.flatMap { $0.next }
}
return result!.next
}
}以上代码也可以从我的 Gist 中找到：https://gist.github.com/tangqiaoboy/af0dc9a45dd309e91436b44b76eab5f1
偷偷告诉你一个小秘密，Gist 里面的代码我稍微修改了两行，最终性能就从打败 LeetCode 10% 的提交变成了打败 LeetCode 50% 的提交，如果你感兴趣，可以自己仔细对比一下。
愿大家玩得开心～
