给定两个用链表表示的整数,每个节点包含一个数位。
这些数位是反向存放的,也就是个位排在链表首部。
编写函数对这两个整数求和,并用链表形式返回结果。
示例:
输入:(7 -> 1 -> 6) + (5 -> 9 -> 2),即617 + 295
输出:2 -> 1 -> 9,即912
进阶:假设这些数位是正向存放的,请再做一遍。
示例:
输入:(6 -> 1 -> 7) + (2 -> 9 -> 5),即617 + 295
输出:9 -> 1 -> 2,即912
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/sum-lists-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
go实现:
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
resHead := &ListNode{0,nil}
p := resHead
flag := 0
for l1 != nil && l2 != nil {
sum := l1.Val + l2.Val + flag
t := &ListNode{0,nil}
if sum >= 10 {
t.Val = sum%10
flag = 1
} else {
flag = 0
t.Val = sum
}
p.Next = t
l1 = l1.Next
l2 = l2.Next
p = p.Next
}
for l1 != nil {
sum := l1.Val + flag
t := &ListNode{0,nil}
if sum >= 10 {
t.Val = sum%10
flag = 1
} else {
flag = 0
t.Val = sum
}
p.Next = t
l1 = l1.Next
p = p.Next
}
for l2 != nil {
sum := l2.Val + flag
t := &ListNode{0,nil}
if sum >= 10 {
t.Val = sum%10
flag = 1
} else {
flag = 0
t.Val = sum
}
p.Next = t
l2 = l2.Next
p = p.Next
}
if flag == 1 {
t := &ListNode{1,nil}
p.Next = t
}
return resHead.Next
}
代码优化:
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
resHead := &ListNode{}
p := resHead
sum := 0
for l1 != nil || l2 != nil || sum != 0 {
if l1 != nil {
sum += l1.Val
l1 = l1.Next
}
if l2 != nil {
sum += l2.Val
l2 = l2.Next
}
p.Next = &ListNode{sum%10,nil}
p = p.Next
sum /= 10
}
return resHead.Next
}