[leetcode]-add two numbers @python

最新推荐文章于 2023-09-21 21:37:40 发布

转载最新推荐文章于 2023-09-21 21:37:40 发布 · 293 阅读

python 同时被 2 个专栏收录

5 篇文章

订阅专栏

leetcode

3 篇文章

订阅专栏

原题地址：https://oj.leetcode.com/problems/add-two-numbers/

题意：

You are given two linked lists representing two non-negative numbers. The digits are stored in reverse order and each of their nodes contain a single digit. Add the two numbers and return it as a linked list.

Input: (2 -> 4 -> 3) + (5 -> 6 -> 4)
Output: 7 -> 0 -> 8

解题思路：链表的操作。

个人代码：

# 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
        """

        #初始化链表，将个位数和放于第一个节点，较为复杂。采用参考代码的方式，第一个节点设为0，返回值为第二个节点
        sum=l1.val+l2.val        if sum<10:            lre=ListNode(sum)            k=0        else:            lre=ListNode(sum%10)            k=1        lr=lre   #保留第一个节点，方便返回           #count=1        while l1.next!=None or l2.next!=None:            l1=l1.next            l2=l2.next            #出现两个加数位数不同的情况            if l1==None:                l1=ListNode(0)             if l2==None:                l2=ListNode(0)            sum=l1.val+l2.val+k            if sum<10 :                lr.next=ListNode(sum)                k=0            else:                lr.next=ListNode(sum%10)                k=1            lr=lr.next            #count+=1        if k==1:            lr.next=ListNode(1)#最高位加法若有进位，要记录        return lre

参考代码：

# Definition for singly-linked list.
# class ListNode:
#     def __init__(self, x):
#         self.val = x
#         self.next = None

class Solution:
    # @return a ListNode
    def addTwoNumbers(self, l1, l2):
        if l1 == None: return l2  #其中一个加数不存在，直接返回存在的加数
        if l2 == None: return l1
        flag = 0
        dummy = ListNode(0); p = dummy #初始化链表，保留链表的一个节点
        while l1 and l2: #两个加数的node均不为空
            p.next = ListNode((l1.val+l2.val+flag) % 10)#不要分和是否大于10，所有的均取10的余数即可
            flag = (l1.val+l2.val+flag) / 10  #记录是否需要进位
            l1 = l1.next; l2 = l2.next; p = p.next 
        if l2:#l1为空，l2不为空的时候
            while l2:
                p.next = ListNode((l2.val+flag) % 10)
                flag = (l2.val+flag) / 10
                l2 = l2.next; p = p.next
        if l1:
            while l1:
                p.next = ListNode((l1.val+flag) % 10)
                flag = (l1.val+flag) / 10
                l1 = l1.next; p = p.next
        if flag == 1: p.next = ListNode(1)
        return dummy.next #返回第一个节点的下一个值

个人改进代码：

# 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
        """
        #考虑只有一个加数的情况
        if l1=None:return l2 
        if l2=None:return l1
        #初始化链表，第一节点为0，返回值为第2个节点
        lre=ListNode(0)
        k=0
        lr=lre#保留第1个节点，方便链表
        #count=1
        while l1.next!=None or l2.next!=None:
            l1=l1.next
            l2=l2.next
            #出现两个加数位数不同的情况
            if l1==None:
                l1=ListNode(0) 
            if l2==None:
                l2=ListNode(0)
            sum=l1.val+l2.val+k
            lr.next=ListNode(sum%10) #节点值为和对10取余
            k=sum/10 #进位为和对10取整
            lr=lr.next
            #count+=1
        if k!=0:
            lr.next=ListNode(1)#最高位加法若有进位，要记录
        return lre.next #从第2个节点开始返回链表

总结：

链表的操作

1）链表类的初始化lre=ListNode(0)  注意参数
第1个节点可以设置为0，只需要从第2个节点开始返回
2）链表的构建
保留链表的第一个节点，使用复制节点开始后面的构建，方便后面对链表整体的操作，例如返回或是打印
#初始化链表，第一节点为0，返回值为第2个节点
        lre=ListNode(0)
        k=0
        lr=lre#保留第1个节点，方便链表
         。。。。。
        return lre.next #从第2个节点开始返回链表