2. 两数相加
给出两个非空的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807
链表是大数做加法的一个很好的数据结构,这里要考虑两种特殊情况:
- 两个链表长度不同
- 两个链表长度相同,但最后一位进位
对于链表长度不同的情况,可以使用:
while l1 or l2:
x = l1.val if l1 else 0
y = l2.val if l2 else 0
也可以使用下述代码的方式,使用三个while循环,在时间和空间复杂度上没有区别。
对于最后一位有进位的情况,要考虑最后的tmp是否为0,不为0则在最后一位加上1即可。
另外要注意的是,链表是单向结构,如果直接return res不能得到结果(None),必须使用head链接到res的头部,最后return head.next。
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def addTwoNumbers(self, l1, l2):
"""
:type l1: ListNode
:type l2: ListNode
:rtype: ListNode
"""
res = ListNode(0)
head = res
tmp = 0
while l1 and l2:
res.next = ListNode((l1.val+l2.val+tmp)%10)
tmp = (l1.val+l2.val+tmp)//10
l1,l2,res = l1.next,l2.next,res.next
while l1:
res.next = ListNode((l1.val+tmp)%10)
tmp = (l1.val+tmp)//10
l1,res = l1.next,res.next
while l2:
res.next = ListNode((l2.val+tmp)%10)
tmp = (l2.val+tmp)//10
l2,res = l2.next,res.next
if tmp:
res.next = ListNode(tmp)
return head.next