LeetCode(2)-Python-两数相加(add-two-numbers)

题目描述:

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

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

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

示例:

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

解题思路:

这是一个用python的单链表的题,最重要的是要注意进位是向右进位,以及注意3位数加上3位数为4位数的情况

# Definition for singly-linked list.
# 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
        """
        
        
        head = ListNode(0)
        new_list = head
        
        flag = 0
        while l1 != None or l2 != None:
            n1 = l1.val if l1 else 0
            n2 = l2.val if l2 else 0
            num = n1 + n2 + flag
            if num >= 10:
                head.next = ListNode(num-10)
                flag = 1
            else:
                head.next = ListNode(num)
                flag = 0
            l1 = l1.next if l1 else None
            l2 = l2.next if l2 else None
            head = head.next
        
        if flag == 1:
            head.next = ListNode(1)
            
        return new_list.next

结果 116ms

你可能感兴趣的:(LeetCode)