leetcode(3)_2_medium_两数相加_python

两数相加

题目描述：
给你两个非空的链表，表示两个非负的整数。它们每位数字都是按照逆序的方式存储的，并且每个节点只能存储一位数字。
请你将两个数相加，并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外，这两个数都不会以 0 开头。
示例 ：
输入：l1 = [2,4,3], l2 = [5,6,4]
输出：[7,0,8]
解释：342 + 465 = 807.
提示：

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

解法

1. 两链表同时从头开始遍历，记录好加和和进位，如果某个链表已经结束，那就补 0 即可。
2. 需要注意，如果最后有进位，需要再开辟 1 个节点，存放进位 1。

代码

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        q1, q2 = l1, l2
        p = ListNode(-1)  # 设置一个 dummy_head，方便存储结果，最后返回它的 next 即可
        q, carry = p, 0  # 初始进位是 0
        while q1 or q2:
            [q1, val1] = [q1.next, q1.val] if q1 else [None, 0]  # 如果到头就认作 0，否则是其 val
            [q2, val2] = [q2.next, q2.val] if q2 else [None, 0]
            q.next = ListNode((val1 + val2 + carry) % 10)
            carry = (val1 + val2 + carry) // 10
            q = q.next
        if carry:
            q.next = ListNode(1)
        return p.next
      

测试结果

执行用时：72 ms, 在所有 Python3 提交中击败了47.60% 的用户
内存消耗：14.8 MB, 在所有 Python3 提交中击败了86.24% 的用户

说明

算法题来源：力扣(LeetCode)