首页 > > 网络编程 > 其它 >

LeetCode链表相加-Python<二>

2018-07-06 01:18:59来源：博客园阅读 ()

上一篇：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