LeetCode链表相加-Python<二>
程序员文章站
2024-01-05 18:03:05
上一篇:LeetCode两数之和-Python<一> 题目:https://leetcode-cn.com/problems/add-two-numbers/description/ 给定两个非空链表来表示两个非负整数。位数按照逆序方式存储,它们的每个节点只存储单个数字。将两数相加返回一个新的链表。 ......
上一篇:
题目:
给定两个非空链表来表示两个非负整数。位数按照逆序方式存储,它们的每个节点只存储单个数字。将两数相加返回一个新的链表。
你可以假设除了数字 0 之外,这两个数字都不会以零开头。
示例:
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4) 输出:7 -> 0 -> 8 原因:342 + 465 = 807
关键点:进位(这里满十加一)、指针移动(这里移向下一个节点)
class Solution: def addTwoNumbers(self, l1, l2): """ :type l1: ListNode :type l2: ListNode :rtype: ListNode """ ans = ListNode(0) #新建一个节点,初始值为0 temp = ans tempsum = 0 while True: if (l1 != None): tempsum = l1.val + tempsum #l1链表节点值添加到总和里 l1 = l1.next #指针指向下一个节点 if (l2 != None): tempsum = tempsum + l2.val #l2链表节点值添加到总和里 l2 = l2.next #指针指向下一个节点 temp.val = tempsum % 10 #取余数(满十进位),赋值当前节点值 print(tempsum) tempsum = int(tempsum / 10) #获取进位数赋值给总和(比如tempsum为10则进1位,否则进位为0),下一次节点相加,从新的总和开始。 if l1 == None and l2 == None and tempsum == 0: #直到没有进位了,同时节点位空了,跳出循环。(这里加上tempsum==0条件是因为,最后两个节 break #点和值可能大于10) temp.next = ListNode(0) #新建下一个节点,存放和 temp = temp.next #指针指向下一个节点 return ans
时间复杂度:O(max(l1长度,l2长度)) ,空间复杂度:O(max(l1长度,l2长度))