题目
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
解决方案
就像在纸上计算两个数字的和那样,我们首先从最低有效位也就是列表 l1 和 l2 的表头开始相加。由于每位数字都应当处于 0…9 的范围内,我们计算两个数字的和时可能会出现“溢出”。例如,5 + 7 = 12。在这种情况下,我们会将当前位的数值设置为 2,并将进位 flag = 1 带入下一次迭代。进位 flag 必定是 0 或 1,这是因为两个数字相加(考虑到进位)可能出现的最大和为 9 + 9 + 1 = 19。
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode { var value, flag int
var l3, cur *ListNode
if l1 == nil && l2 == nil {
return initListNode(0)
}
for {
sum := 0
if l1 == nil && l2 == nil {
if flag != 0 {
cur.Next = initListNode(flag)
}
return l3
}
if l1 != nil {
sum += l1.Val
l1 = l1.Next
}
if l2 != nil {
sum += l2.Val
l2 = l2.Next
}
if flag != 0 {
sum += flag
}
value = sum % 10
flag = sum / 10
if l3 != nil {
cur.Next = initListNode(value)
cur = cur.Next
} else {
l3 = initListNode(value)
cur = l3
}
}
}
func initListNode(n int) *ListNode {
node := &ListNode{}
node.Val = n
node.Next = nil
return node
}
复杂度分析
- 时间复杂度:O(max(m, n)),假设 m 和 n 分别表示 l1l1 和 l2 的长度,上面的算法最多重复 max(m, n) 次。
- 空间复杂度:O(max(m, n)), 新列表的长度最多为 max(m,n) + 1max(m,n)+1。