[LeetCode] 两数相加

2018-11-03  本文已影响0人  数据麻瓜

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

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

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

考察点

  1. 如何进位 -> 引入新变量来记录
  2. 链表长度不一致 -> 另长度短的链表之后都取0

Pitfall:

输入:(5) + (5)
正确输出:0 -> 1
错误输出:0

一开始在while的循环条件中没有考虑到两个链表同时结束但还需进位的问题,导致出错

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

class Solution:
    def addTwoNumbers(self, l1, l2):
        dummy = 0   #判断是否需要进位
        output_temp = output = ListNode(0)  #创建返回的链表,output用于循环迭代,_temp用于输出
        while (l1 or l2 or dummy>0):  #如果l1,l2有一个没完或仍需要进位,则继续循环
            v1 = v2 = 0  #为了保证可加性
            if l1:
                v1 = l1.val
                l1 = l1.next
            if l2:
                v2 = l2.val
                l2 = l2.next
            temp = v1 + v2
            if dummy > 0 :  #若需进位
                temp = temp + dummy  #值+1
            output.next = ListNode(temp % 10) #取余
            output = output.next #建下一个节点
            dummy = temp // 10 #下一位是否要进位
        return(output_temp.next)

结果
Runtime: 112 ms, faster than 81.72% of Python3 online submissions for Add Two Numbers.
时间复杂度:O(max(n,m))

Notes: Python中next的使用

上一篇下一篇

猜你喜欢

热点阅读