题目在leetcode上的链接为:
https://leetcode-cn.com/problems/add-two-numbers/
题目描述
解题思路
将链表中对应节点数字进行相加的操作,同时注意考虑进位问题即可。具体的做法如下:
初始化进位数carry=0
初始化p1,p2分别为l1,l2的头节点
循环判断当p1或者p2不为空时:
如果p1不为空,则将p1节点的值赋给x,如果p1为空,则进行补零,将0赋给x
如果p2不为空,则将p2节点的值赋给y,如果p2为空,则进行补零,将0赋给y
此时的value值为x+y+carry
如果value大于9,则将value置为value-10且进位数carry置为1;否则将carry置为0
将value作为值创建一个新的节点插入到新链表的尾部
判断p1,p2是否为空,如果不为空则前进一个节点
循环结束后判断进位数是否大于0,若大于0,则在新链表的尾部增加一个值为carry的节点
解题思路中需要注意的两点为:
(1)当p1或者p2为空时,要进行补零操作
(2)最后可能会存在进位的情况,需要考虑
复杂度分析:
设m,n分别表示链表l1,l2的长度
由于需要遍历完两个链表,所以该算法的时间复杂度为o(max(m,n))
由于需要创建一个新的链表储存结果,该算法的空间复杂度为o(max(m,n)),需要创建的链表的最大长度为o(max(m,n))+1
python代码:
# Definition for singly-linked list.
# class ListNode(object):
# def __init__(self, x):
# self.val = x
# self.next = None
class Solution(object):
def addTwoNumbers(self, l1, l2):
"""
:type l1: ListNode
:type l2: ListNode
:rtype: ListNode
"""
p1 = l1
p2 = l2
head = None
current = None
carry = 0 #进位数
while p1 != None or p2 != None:
x = p1.val if p1 != None else 0
y = p2.val if p2 != None else 0
value = x + y + carry
if value > 9:
value = value - 10
carry = 1
else:
carry = 0
if head == None:
head = ListNode(value)
current = head
else:
current.next = ListNode(value)
current = current.next
if p1 != None:
p1 = p1.next
if p2 != None:
p2 = p2.next
if carry > 0:
current.next = ListNode(carry)
return head