首页 > 编程语言 > 详细

LeetCode in Python 2. 两数相加

时间:2019-08-18 01:49:21      阅读:123      评论:0      收藏:0      [点我收藏+]

给出两个 非空 的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。

如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。

您可以假设除了数字 0 之外,这两个数都不会以 0 开头。

示例:

输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807

 

Solution:

注意最后一个进位是0的时候需要把它设为None,另外每次新建next节点时,初始值应设为carry而不是0,例如:5 + 7 = 2->1

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

        head, carry = ListNode(0), 0
        l3 = prev = head
        while l1 and l2:
            sum = l1.val + l2.val + carry
            l3.val, carry = sum % 10, sum//10
            l3.next = ListNode(carry) # e.g. 5, 7 = 2->1
            prev = l3
            l1, l2, l3 = l1.next, l2.next, l3.next
            
        l4 = l1 if l1 else l2
        
        while l4:
            sum = l4.val + carry
            l3.val, carry = sum % 10, sum//10
            l3.next = ListNode(carry)
            prev = l3
            l3, l4 = l3.next, l4.next
            
        if not carry:
            prev.next = None
        
        return head

LeetCode in Python 2. 两数相加

原文:https://www.cnblogs.com/lowkeysingsing/p/11370992.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!