(python刷题)leetcode第2题:两数相加

题目在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

发布了37 篇原创文章 · 获赞 6 · 访问量 5401

猜你喜欢

转载自blog.csdn.net/qq_37891889/article/details/104155009