LeetCode:2. 两数相加(python)
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
思路:
- 加法规则
- 注意进位符和边界的处理
- 遍历结束时,注意剩余进位符处理
附代码(Python):
class ListNode:
def __init__(self, x):
self.val = x
self.next = None
class Solution:
def addTwoNumbers(self, l1, l2):
if not l1: return l2
if not l2: return l1
res = cur = ListNode(None)
carry = 0 # 进位符
while l1 or l2:
# 遍历链表,计算当前位数值,移动指针
if not l1:
sum_ = l2.val+carry
l2 = l2.next
elif not l2:
sum_ = l1.val+carry
l1 = l1.next
else:
sum_ = l1.val+l2.val+carry
l1, l2 = l1.next, l2.next
carry = sum_//10 # 更新进位符
cur.next = ListNode(sum_%10) # 新建节点值
cur = cur.next
if carry: # 若剩余进位符,则添加节点
cur.next = ListNode(carry)
return res.next
# 构建链表
def build_link(nums):
li = cur = ListNode(None)
for i in nums:
cur.next = ListNode(i)
cur = cur.next
return li.next
nums1 = [2, 4, 3]
nums2 = [5, 6, 4]
l1 = build_link(nums1)
l2 = build_link(nums2)
# 两数相加
test = Solution()
res = test.addTwoNumbers(l1, l2)
# 打印
while res:
print(res.val, end='')
res = res.next
print()
708