一、题目描述
给出两个非空的链表用来表示两个非负的整数。其中,它们各自的位数是按照逆序的方式存储的,并且它们的每个节点只能存储一位数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
二、求解
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
# 创建头节点,方便最后返回
pre_head = ListNode(-1)
head = pre_head
# 定义进位
carry = 0
while l1 or l2:
if l1:
l1_val = l1.val
else:
l1_val = 0
if l2:
l2_val = l2.val
else:
l2_val = 0
curr_sum = l1_val + l2_val + carry
carry = curr_sum // 10
# 创建当前节点的下一个节点——即下一位
head.next = ListNode(curr_sum%10)
head = head.next
if l1:
l1 = l1.next
if l2:
l2 = l2.next
# 最后还需要单独看看是否有新进位
if carry:
head.next = ListNode(1)
return pre_head.next