题目
有两个非空的链表,每个链表代表一个非负整数。这些数字的位数是以逆序存储的(即个位在链表头部),并且每个节点包含一个单独的数字。你的任务是将这两个数字相加,并将结果以同样的链表形式返回。
示例:
Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8
Explanation: 342 + 465 = 807.
解题思路
为了处理⽅法统⼀,可以先建⽴⼀个虚拟头结点,这个虚拟头结点的 Next 指向真正的 head,这样head 不需要单独处理,直接 while 循环即可。另外判断循环终⽌的条件不⽤是 p.Next != nil,这样最后⼀位还需要额外计算,循环终⽌条件应该是 p != nil。
答案:(Go语言)
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
if l1 == nil || l2 == nil {
return nil
}
head := &ListNode{Val: 0, Next: nil}
current := head
carry := 0
for l1 != nil || l2 != nil {
var x, y int
if l1 == nil {
x = 0
} else {
x = l1.Val
}
if l2 == nil {
y = 0
} else {
y = l2.Val
}
current.Next = &ListNode{Val: (x + y + carry) % 10, Next: nil}
current = current.Next
carry = (x + y + carry) / 10
if l1 != nil {
l1 = l1.Next
}
if l2 != nil {
l2 = l2.Next
}
}
if carry > 0 {
current.Next = &ListNode{Val: carry % 10, Next: nil}
}
return head.Next
}