题目
您将获得两个非空链接列表,表示两个非负整数。 数字以相反的顺序存储,并且它们的每个节点包含单个数字。 添加两个数字并将其作为链接列表返回。
您可以假定这两个数字不包含任何前导零,除了数字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)")