目录
一、题目:
给你两个 非空 的链表,表示两个非负的整数。它们每位数字都是按照 逆序 的方式存储的,并且每个节点只能存储 一位 数字。
请你将两个数相加,并以相同形式返回一个表示和的链表。
你可以假设除了数字 0 之外,这两个数都不会以 0 开头。
提示:
- 每个链表中的节点数在范围
[1, 100]
内 0 <= Node.val <= 9
- 题目数据保证列表表示的数字不含前导零
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/add-two-numbers
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二、解法
解法一:
分为两种情况讨论,分别是和两个链表等长,与两个链表其中一个更长
只遍历了一遍两个链表,时间复杂度为O(n)
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
"""
backup_l1=l1
backup_l2=l2
isOverWhelm=False
##首先遍历到两个链表的公共长度
while(l1.next!=None and l2.next!=None):
if(isOverWhelm):
l1.val+=1
isOverWhelm=False
if(l1.val+l2.val>=10):
isOverWhelm=True
l1.val=(l1.val+l2.val)%10
l2.val=l1.val
l1=l1.next
l2=l2.next
##若有一个链表更长则为target
if(l1.next!=None):
target=l1
result=1
elif(l2.next!=None):
target=l2
result=2
#若两个链表等长
else:
target=l1
target.val=l1.val+l2.val
if(isOverWhelm):
target.val+=1
if(target.val==10):
target.val=0
target.next=ListNode(1)
elif(target.val>10):
target.val=target.val%10
target.next=target.next=ListNode(1)
return backup_l1
target.val=l1.val+l2.val
#遍历完target的剩下节点
while(target.next!=None):
if(isOverWhelm):
target.val+=1
isOverWhelm=False
if(target.val>=10):
isOverWhelm=True
target.val=target.val%10
target=target.next
if(isOverWhelm):
#此时只可能通过低位的进一来+1产生进位
if(target.val+1>=10):
target.val=0
target.next=ListNode(1)
else:
target.val+=1
if(result==1):
return backup_l1
else:
return backup_l2
总结:
要善于利用分类思想