两数相加
2019-02-22 本文已影响0人
scriptllh
leetocde 2
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
/**
* Definition for singly-linked list.
* type ListNode struct {
* Val int
* Next *ListNode
* }
*/
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
if l1 == nil {
return l2
}
if l2 == nil{
return l1
}
res := &ListNode{}
cur := res
for l1 != nil && l2 != nil{
sum := l1.Val + l2.Val
if sum >= 10{
sum %= 10
if l1.Next == nil{
l1.Next = &ListNode{}
}
l1.Next.Val += 1
if l1.Next.Val >= 10{
if l2.Next == nil{
l2.Next = &ListNode{}
}
}
}
cur.Next = &ListNode{
Val: sum,
}
l1 = l1.Next
l2 = l2.Next
cur = cur.Next
}
for l1 != nil{
cur.Next = l1
l1 = l1.Next
cur = cur.Next
}
for l2 != nil{
cur.Next = l2
l2 = l2.Next
cur = cur.Next
}
return res.Next
}