跳转至

2. 两数相加

题目

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

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

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

示例 1:

输入：l1 = [2,4,3], l2 = [5,6,4]

输出：[7,0,8]

示例 2:

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

输出：[0]

题解

Go
func addTwoNumbers(l1, l2 *ListNode) *ListNode {
    dummy := &ListNode{} // 哨兵节点
    curr := dummy

    carry := 0 // 进位
    for l1 != nil || l2 != nil || carry != 0 { // 有一个不是空节点，或者还有进位，就继续迭代
        if l1 != nil {
            carry += l1.Val // 节点值和进位加在一起
            l1 = l1.Next // 下一个节点
        }
        if l2 != nil {
            carry += l2.Val // 节点值和进位加在一起
            l2 = l2.Next // 下一个节点
        }
        curr.Next = &ListNode{Val: carry % 10} // 每个节点保存一个数位
        carry /= 10 // 新的进位
        curr = curr.Next // 下一个节点
    }
    return dummy.Next // 哨兵节点的下一个节点就是头节点
}
Python
class Solution:
    def addTwoNumbers(self, l1: Optional[ListNode], l2: Optional[ListNode]) -> Optional[ListNode]:
        cur = dummy = ListNode()  # 哨兵节点
        carry = 0  # 进位
        while l1 or l2 or carry:  # 有一个不是空节点，或者还有进位，就继续迭代
            if l1:
                carry += l1.val  # 节点值和进位加在一起
                l1 = l1.next  # 下一个节点
            if l2:
                carry += l2.val  # 节点值和进位加在一起
                l2 = l2.next  # 下一个节点
            cur.next = ListNode(carry % 10)  # 每个节点保存一个数位
            carry //= 10  # 新的进位
            cur = cur.next  # 下一个节点
        return dummy.next  # 哨兵节点的下一个节点就是头节点

复杂度

双指针+虚拟头结点

  • 时间复杂度：\(O(n)\)
  • 空间复杂度：\(O(1)\)

参考