题目
https://leetcode-cn.com/problems/add-two-numbers/
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
思路
考虑用变量carry,保存进位。
长度不一致的话,短的链表后面加上0替代加上原来链表结点的值。
代码
# 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:
# 用一个新的链表表示进位
ans = dummy_node = ListNode(0) # inplace
tmp_node = ListNode(0)
carry = 0
while l1 or l2:
x = l1.val if l1 else 0
y = l2.val if l2 else 0
tmp_sum = x + y + carry
dummy_node.next = ListNode(tmp_sum % 10)
carry = tmp_sum // 10
dummy_node = dummy_node.next
if l1: l1 = l1.next
if l2: l2 = l2.next
if carry > 0:
dummy_node.next = ListNode(carry)
return ans.next