LEETCODE 2. 两数相加

LEETCODE 2. 两数相加

题目描述

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

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

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

示例:

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

题目地址
中文版
英文版

代码实现

class Solution(object):
    def addTwoNumbers(self, l1, l2):
        """
        :type l1: ListNode
        :type l2: ListNode
        :rtype: ListNode
        """
        if not l1 or not l2:
            return l1 if l1 else l2
        a,b = l1,l2
        p = a
        carry = 0
        while a and b:
            val = a.val+b.val+carry
            if val>=10:
                carry = val/10
                val %= 10
            else:
                carry = 0
            p = a
            p.val = val
            a = a.next
            b = b.next
        q = a if a else b
        if q:
            p.next = q
            while q:
                val = q.val+carry
                if val>=10:
                    carry = val/10
                    val %= 10
                else:
                    carry = 0
                q.val = val
                p = q
                q = q.next   
        if carry:
            p.next = ListNode(carry)
        return l1 
                                                                 

极简的代码

class Solution(object):
    def addTwoNumbers(self, l1, l2):
        a,b,p,carry = l1,l2,None,0
        while a or b:
            val = (a.val if a else 0)+(b.val if b else 0)+carry
            carry,val = val/10 if val>=10 else 0,val%10
            p,p.val = a if a else b,val
            a,b = a.next if a else None,b.next if b else None
            p.next = a if a else b
        if carry:
            p.next = ListNode(carry)
        return l1
                                                                    
3 次阅读

发表评论

电子邮件地址不会被公开。