【题目】
(中)给出两个非空的链表用来表示两个非负的整数。其中,它们各自的位数是按照逆序 的方式存储的,并且它们的每个节点只能存储一位数字。如果我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
【示例】
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
【解答】
这道题目就是让实现小学加法的,不同的是用到了链表的数据结构,需要注意的是,链表顺序与数字各个位的先后顺序恰好相反,链表的第一个数字代表个位,依次往下。因此计算时,可以按照链表顺序直接计算相加结果。需要注意的是,两个链表的长度可能不同,而且可能存在链表为空的情况。
这里写一段伪代码说明流程:
def 两数相加(头结点1, 头结点2):
定义一个临时结点pre_head,其后挂载计算结果;
定义结点cur,记录用于当前位的计算结果,并设置为pre_head;
当两个链表的当前结点中有一个不为空,则执行:
取两个加数链表的当前位,如果为空设置为0;
计算当前位的相加结果r和进位结果c(缓存,下次迭代使用);
将计算结果作为该位结点挂在当前结果链表末尾cur中,并将cur后移;
如果链表1当前结点不为空,则后移,链表2亦是如此。
判断是否还有进位c,如果为1则结果链表增加一个值为1的节点作为最高位
return 临时结点pre_head的下一个结点
# (pre_head只是为了构造结果链表,并非结果的组成部分)
具体代码实现:
# 定义一个链表节点
class ListNode(object):
def __init__(self, x):
self.val = x
self.next = None
# 这里为大家补充了创建链表的函数
def create_list(nums):
last = None
for num in reversed(nums):
list_node = ListNode(num)
list_node.next = last
last = list_node
return last
# 解决方案
class Solution(object):
def addTwoNumbers(self, l1, l2):
re = ListNode(0)
r = re
carry = 0 # 进位,0或1
while l1 and l2:
a = l1.val if l1 else 0
b = l2.val if l2 else 0
s = carry + a + b
carry = s // 10 # 求进位
s = s % 10 # 求余数
r.next = ListNode(s) #
r = r.next
if l1 is not None:
l1 = l1.next
if l2 is not None:
l2 = l2.next
if carry > 0:
r.next = ListNode(1)
return re.next
if __name__ == "__main__":
l1 = create_list([3, 4, 7])
l2 = create_list([6, 4])
s = Solution()
print(s.addTwoNumbers(l1, l2))
如果您有任何疑问,请在评论区留言。