leetcode No.2 两数相加 (python3实现)

Posted 2021-03-15 everfight的成长之路

来源

https://leetcode-cn.com/problems/add-two-numbers/

题目描述

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

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

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

示例：

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

代码实现

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

class Solution:
    def addTwoNumbers(self, l1: ListNode, l2: ListNode) -> ListNode:
        n = l1.val + l2.val 
        l3 = ListNode(n % 10)
        l3.next = ListNode(n // 10)
        p1 = l1.next 
        p2 = l2.next 
        p3 = l3 
        while True:
            if p1 and p2:
                total = p1.val + p2.val + p3.next.val 
                p3.next.val = total % 10
                p3.next.next = ListNode(total // 10)
                p1 = p1.next 
                p2 = p2.next
                p3 = p3.next
            elif p1 and not p2:
                total = p1.val + p3.next.val 
                p3.next.val = total % 10
                p3.next.next = ListNode(total // 10)
                p1 = p1.next 
                p3 = p3.next
            elif not p1 and p2:
                total = p2.val + p3.next.val 
                p3.next.val = total % 10
                p3.next.next = ListNode(total // 10) 
                p2 = p2.next
                p3 = p3.next
            else:
                if p3.next.val == 0:
                    p3.next = None
                break
        return l3

核心思想就是设置一个新的变量l3,判断它是否大于10，然后考虑四种情况，

P1 P2同时存在，
P1存在，P2不存在，
P2存在，P1不存在，
P1 P2都不存在

然后计算后不断的更新next节点，最终输出该变量。