两数相加
题目描述:
给你两个非空的链表,表示两个非负的整数。它们每位数字都是按照逆序的方式存储的,并且每个节点只能存储一位数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例 :
输入:l1 = [2,4,3], l2 = [5,6,4]
输出:[7,0,8]
解释:342 + 465 = 807.
提示:
- 每个链表中的节点数在范围 [1, 100] 内
- 0 <= Node.val <= 9
- 题目数据保证列表表示的数字不含前导零
解法
- 两链表同时从头开始遍历,记录好加和和进位,如果某个链表已经结束,那就补 0 即可。
- 需要注意,如果最后有进位,需要再开辟 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)