两数相加

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
}
上一篇下一篇

猜你喜欢

热点阅读