【Leetcode】2 addTwoNumbers python解法

给定两个非空链表来表示两个非负整数。位数按照逆序方式存储,它们的每个节点只存储单个数字。将两数相加返回一个新的链表。

你可以假设除了数字 0 之外,这两个数字都不会以零开头。
输入:(2 -> 4 -> 3) + (5 -> 6 -> 4)
输出:7 -> 0 -> 8
原因:342 + 465 = 807

python解法
思路简单使用 flag 标记考虑进位的问题直接相加,因为要返回头结点所以事先做一个哑结点指向下一个,最后返回哑结点的next就好了

class ListNode:
    def __init__(self, x):
        self.val = x
        self.next = None

class Solution:
    def addTwoNumbers(self, l1, l2):
        """
        :type l1: ListNode
        :type l2: ListNode
        :rtype: ListNode
        """
        flag = 0
        res = ListNode(0)
        cur = res
        while l1 or l2:
            a1 = l1.val if l1 else 0
            a2 = l2.val if l2 else 0
            s = a1 + a2 + flag
            if s > 9:
                s -= 10
                flag=1
            else:
                flag=0
            if l1:l1 = l1.next
            if l2:l2 = l2.next
            cur.next = ListNode(s)
            cur = cur.next
        if flag == 1:
            cur.next = ListNode(1)
        return res.next

你可能感兴趣的:(Leetcode)