字节跳动面试真题-两数相加

题目描述

给你两个 非空 的链表，表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的，并且每个节点只能存储 一位 数字。

请你将两个数相加，并以相同形式返回一个表示和的链表。

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

示例1：

输入：l1 = [2,4,3], l2 = [5,6,4]
输出：[7,0,8]
解释：342 + 465 = 807.

示例2：

输入：l1 = [0], l2 = [0]
输出：[0]

示例3：

输入：l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9]
输出：[8,9,9,9,0,0,0,1]

提示：
每个链表中的节点数在范围 [1, 100] 内
0 <= Node.val <= 9
题目数据保证列表表示的数字不含前导零

题解

思路

这道中等难度的题目最近半年，在外企面试的一些题目中经常出现，例如亚马逊、微软、Facebook，其中字节跳动也在多场面试中出了这道题目。

这是个典型的链表类的题目

首先我们将两个链表看成是相同长度的进行遍历，如果一个链表较短则在前面补 0，比如 示例3，9999999+0009999= 10009998，每一位计算的同时需要考虑上一位的进位问题，而当前位计算结束后同样需要更新进位值，如果两个链表全部遍历完毕后，进位值为 1，则在新链表最前方添加节点 1。

具体来说就是，如果当前两个链表处相应位置的数字为 n1,n2，进位值为 $\text{carry}$，则它们的和为 n1+n2+$\text{carry}$；其中，答案链表处相应位置的数字为 $(n1+n2+\text{carry})\mathrm{m}\mathrm{o}\mathrm{d}10$，而新的进位值为$\frac{n1+n2+\text{carry}}{10}$。

我们在解链表类问题的时候，如果返回结果为头结点时，通常需要先初始化一个预先指针 pre，该指针的下一个节点指向真正的头结点head。

使用预先指针的目的在于避免链表初始化时无可用节点值，而且链表构造过程需要指针移动，进而会导致头指针丢失，无法返回结果。

代码

Go

func addTwoNumbers(l1, l2 *ListNode) (head *ListNode) {
    var tail *ListNode
    carry := 0
    for l1 != nil || l2 != nil {
        n1, n2 := 0, 0
        if l1 != nil {
            n1 = l1.Val
            l1 = l1.Next
        }
        if l2 != nil {
            n2 = l2.Val
            l2 = l2.Next
        }
        sum := n1 + n2 + carry
        sum, carry = sum%10, sum/10
        if head == nil {
            head = &ListNode{Val: sum}
            tail = head
        } else {
            tail.Next = &ListNode{Val: sum}
            tail = tail.Next
        }
    }
    if carry > 0 {
        tail.Next = &ListNode{Val: carry}
    }
    return
}

复杂度分析

• 时间复杂度：$O(\max (m,n))$，其中 m 和 n 分别为两个链表的长度。我们要遍历两个链表的全部位置，而处理每个位置只需要 O(1)的时间。

• 空间复杂度：O(1)。注意返回值不计入空间复杂度。