Leetcode 2.两数相加(medium)

题目

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

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

你可以假设除了数字 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]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/add-two-numbers
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路

1. 原始解法

 建立一个新的链表存储答案。首先按位遍历l1和l2，用变量add记录是否有进位。如果两数之和大于10，则有进位。

v1 = l1.val
v2 = l2.val
v_sum = v1 + v2 + add
add = 0
if v_sum >= 10:
    v_sum -= 10
    add = 1

当l1或者l2被读完时，将另外一个链表剩下的元素接入存储答案的链表（仍使用加法而非直接合并）。

最后需要注意的是如果所有元素都被读完，但仍有进位，即add为1，需要在结尾再添加一个节点。

结果：通过测试用例，用时52ms，内存使用量15MB。

完整代码：

class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        add = 0
        answer = ListNode(0)
        p = answer
        while l1 and l2:
            v1 = l1.val
            v2 = l2.val
            v_sum = v1 + v2 + add
            add = 0
            if v_sum >= 10:
                v_sum -= 10
                add = 1
            
            p.next = ListNode(v_sum)
            p = p.next

            l1 = l1.next
            l2 = l2.next
            
        if l1:
            while l1:
                v_sum = l1.val + add
                add = 0
                if v_sum >= 10:
                    v_sum -= 10
                    add = 1
                p.next = ListNode(v_sum)
                p = p.next

                l1 = l1.next

        if l2:
            while l2:                
                v_sum = l2.val + add
                add = 0
                if v_sum >= 10:
                    v_sum -= 10
                    add = 1
                p.next = ListNode(v_sum)
                p = p.next

                l2 = l2.next

        if add:
            p.next = ListNode(1)

        return answer.next

2. 第一次优化

注意到其实三段while循环有很大的重复度，可以将三者合并。

当l1或者l2有元素时，读取元素，将指针指向下一个节点，同时将另外一个元素设置为0。

结果：通过测试用例，用时52ms，内存使用量14.8MB

代码：

class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        add = 0
        answer = ListNode(0)
        p = answer
        while l1 or l2:
            v1 = 0
            v2 = 0
            if l1:
                v1 = l1.val
                l1 = l1.next
            if l2:
                v2 = l2.val
                l2 = l2.next
            v_sum = v1 + v2 + add
            add = 0
            if v_sum >= 10:
                v_sum -= 10
                add = 1
            
            p.next = ListNode(v_sum)
            p = p.next

        if add:
            p.next = ListNode(1)

        return answer.next

相关知识

1. 链表