LeetCode菜鸟从零开始——第二题:两数相加
问题描述
给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
问题解答(python)
class Solution:
def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
a = l1
b = l2
p = ListNode(0)
q = p
flag = 0
while a or b or flag:#一定不要忘了flag,否则5+5=0
val = (a.val if a else 0) + (b.val if b else 0) + flag
val,flag = val % 10,1 if val >= 10 else 0
q.val = val
if (a.next if a else a) or (b.next if b else b) or flag:#同上
q.next = ListNode(0)
q = q.next
if a:
a = a.next
if b:
b = b.next
return p
问题总结
借着这道题可以同时回忆了一下大学期间c语言的链表知识
两个链表相加,我们用一个新的链表来保存相加的和,于是新建一个链表,p指向链表头,q是用来移动到下一节点的指针。
这是一道加法题,那就要考虑进位,也就是flag的作用。
以一个两位数和一个三位数相加为例,结果可能是三位数也可能是四位数,计算过程是这样的:
1.个位数字相加
2.十位数字相加,并与进位相加(也可能没有)
3.三位数的百位数字与进位相加(也可能没有)
4.如果有进位那千位为1
也就得到了我们的while循环的循环条件a不为空或b不为空或flag不为0,满足条件的都要继续加下去。
下面介绍一个Python中的小语法
a if condition else b。当满足条件condition时,执行a,否则,执行b。