【力扣-每日小练】2. 两数相加 （python)

2. 两数相加

题目链接：https://leetcode-cn.com/problems/add-two-numbers/

题目描述

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

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

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

示例

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

输入：(0) + (0)
输出：0
原因：0 + 0= 0

输入：(9 -> 8) + (1)
输出：0 -> 9
原因：89 + 1 = 90

# 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:
        
        result = ListNode(0)
        curr = result
        flag = 0
        while l1 or l2:
            temp1 = 0
            temp2 = 0
            if l1:
                temp1 =  l1.val
                l1 = l1.next
            if l2:
                temp2 = l2.val
                l2 = l2.next
            temp = temp1 + temp2 + flag
            curr.next = ListNode(temp%10)
            curr = curr.next
            flag = temp // 10
        
        if flag != 0:
            curr.next = ListNode(1)

        return result.next if result.next else result