Leetcode解题之路(golang版):2. 两数相加(Add Two Numbers)

2. 两数相加

给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。

如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。

您可以假设除了数字 0 之外,这两个数都不会以 0 开头。

示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807

解法一:

整体的思路为先把对应的val值相加,然后得到一个链表,这里面可能会存在val > 10 的情况,没关系,我们再便利一次,将val > 10的处理一下。

空间复杂度:$O(1)$

时间复杂度:$O(n)$

/**
 * Definition for singly-linked list.
 * type ListNode struct {
 *     Val int
 *     Next *ListNode
 * }
 */
func addTwoNumbers(l1 *ListNode, l2 *ListNode) *ListNode {
    if l1 == nil && l2 == nil {
        return l1
    }
    //pre记录了两个链表的结点head
    prel1 := l1
    prel2 := l2
    //flag标记了最长的链表是否为l1
    flag := true
    for l1 != nil || l2 != nil {
        if l1 != nil && l2 != nil {
        //对于这种情况,需要将val值相加,由于不确定最长的链表是哪一个,将两个都更新
            l1.Val = l1.Val + l2.Val
            l2.Val = l1.Val
            l1 = l1.Next
            l2 = l2.Next
        } else if l1 != nil {
            break
        } else {
            flag = false
            break
        }
    }
    now := prel1
    //如果最长的链表不是l1,需要将now更新为l2
    if !flag {
        now = prel2
    }
    prel1 = now
    //处理val大于10的情况
    for now.Next != nil {
        if now.Val + now.Next.Val < 10 {
            now = now.Next
            continue
        }
        now.Next.Val = now.Val / 10 + now.Next.Val
        now.Val = now.Val % 10
        now = now.Next
    }
    //如果最后一个结点val也大于10,需要新增一个结点
    if now.Val >= 10 {
        var next *ListNode
        next = new(ListNode)
        next.Next = nil
        next.Val = now.Val / 10
        now.Val %= 10
        now.Next = next
    }
    return prel1
}

你可能感兴趣的:(leetcode)