题目:
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外,这两个数都不会以 0 开头
输入:l1 = [2,4,3], l2 = [5,6,4] 输出:[7,0,8] 解释:342 + 465 = 807.
示例 2:
输入:l1 = [0], l2 = [0] 输出:[0]
示例 3:
输入:l1 = [9,9,9,9,9,9,9], l2 = [9,9,9,9] 输出:[8,9,9,9,0,0,0,1]
提示:
- 每个链表中的节点数在范围
[1, 100]
内 0 <= Node.val <= 9
- 题目数据保证列表表示的数字不含前导零
Solution:
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, val=0, next=None):
# self.val = val
# self.next = next
class ListNode(object):
def __init__(self, val=0, next=None):
self.val = val
self.next = next
class Solution(object):
def addTwoNumbers(self, l1, l2):
"""
:type l1: ListNode
:type l2: ListNode
:rtype: ListNode
"""
head = p = ListNode(None)
head.next = p
extra = 0
while l1 and l2 :#当两个数字都还有的时候
p.next = ListNode((l1.val+extra+l2.val)%10)
p = p.next
if(l1.val+l2.val+extra>=10):#查看是否有进位
extra = 1
else:
extra = 0
l1 = l1.next
l2 = l2.next
#跳出循环,说明至少有一个链表空了
while l1!=None:
p.next = ListNode((l1.val+extra)%10)
p = p.next
if(l1.val+extra>=10):#查看是否有进位
extra = 1
else:
extra = 0
l1 = l1.next
while l2!=None:
p.next = ListNode((l2.val+extra)%10)
p = p.next
if(l2.val+extra>=10):#查看是否有进位
extra = 1
else:
extra = 0
l2 = l2.next
if extra:
p.next = ListNode(extra)
return head.next
刚开始做的时候我以为最后输出的是个数组,后来报错了才反应过来
结果上看还挺对的,也很快,但是感觉这个方法比较冗余。
Solution2:
之后在答案里看到了更简介的解答:
# Definition for singly-linked list.
# class ListNode:
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution:
def addTwoNumbers(self, l1, l2):
# 创建一个结点值为 None 的头结点, dummy 和 p 指向头结点, dummy 用来最后返回, p 用来遍历
dummy = p = ListNode(None)
s = 0 # 初始化进位 s 为 0
while l1 or l2 or s:
# 如果 l1 或 l2 存在, 则取l1的值 + l2的值 + s(s初始为0, 如果下面有进位1, 下次加上)
s += (l1.val if l1 else 0) + (l2.val if l2 else 0)
p.next = ListNode(s % 10) # p.next 指向新链表, 用来创建一个新的链表
p = p.next # p 向后遍历
s //= 10 # 有进位情况则取模, eg. s = 18, 18 // 10 = 1
l1 = l1.next if l1 else None # 如果l1存在, 则向后遍历, 否则为 None
l2 = l2.next if l2 else None # 如果l2存在, 则向后遍历, 否则为 None
return dummy.next # 返回 dummy 的下一个节点, 因为 dummy 指向的是空的头结点, 下一个节点才是新建链表的后序节点
这里最有趣的地方在于用了个三目运算符就把代码量降低了不少。当有一个链表空了就不再继续遍历后序
while l1 or l2 or s
这个循环条件就保证了无论是l1,l2还是有进位都要继续运行
像别人学习