2018-12-13 LeetCode Q2 两数相加

2. 两数相加

给出两个非空的链表用来表示两个非负的整数。其中,它们各自的位数是按照 逆序 的方式存储的,并且它们的每个节点只能存储 一位 数字。
如果,我们将这两个数相加起来,则会返回一个新的链表来表示它们的和。
您可以假设除了数字 0 之外,这两个数都不会以 0 开头。

示例:

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

链表是大数做加法的一个很好的数据结构,这里要考虑两种特殊情况:

  • 两个链表长度不同
  • 两个链表长度相同,但最后一位进位

对于链表长度不同的情况,可以使用:

while l1 or l2:
	x = l1.val if l1 else 0
	y = l2.val if l2 else 0

也可以使用下述代码的方式,使用三个while循环,在时间和空间复杂度上没有区别。

对于最后一位有进位的情况,要考虑最后的tmp是否为0,不为0则在最后一位加上1即可。

另外要注意的是,链表是单向结构,如果直接return res不能得到结果(None),必须使用head链接到res的头部,最后return head.next。

# 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
        """
        res = ListNode(0)
        head = res
        tmp = 0
        while l1 and l2:
            res.next = ListNode((l1.val+l2.val+tmp)%10)
            tmp = (l1.val+l2.val+tmp)//10
            l1,l2,res = l1.next,l2.next,res.next
        while l1:
            res.next = ListNode((l1.val+tmp)%10)
            tmp = (l1.val+tmp)//10
            l1,res = l1.next,res.next
        while l2:
            res.next = ListNode((l2.val+tmp)%10)
            tmp = (l2.val+tmp)//10
            l2,res = l2.next,res.next
        if tmp:
            res.next = ListNode(tmp)
        return head.next

你可能感兴趣的:(Algorithm,Leetcode,Python)