python(leetcode)-66加一问题

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。

最高位数字存放在数组的首位， 数组中每个元素只存储一个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

　看到题目觉得非常简单 只需要找到list的最后一个下标位置使其加一 即可。

然而天真如我 下面情况才是这题的重点

输入: [9,9,9]
输出: [1,0,0,0]
解释: 输入数组表示数字 999。

　需要考虑进位的情况

重新思考：

笔者想到的是先将数组数字转为十进制数字，然后加一，接着按位取模得到每一位的值。

上代码（花费时间48ms)击败70%

import math
class Solution:
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        a=1
        num=0
        lens=len(digits)
        for i in range(lens-1,-1,-1):
            num=num+digits[i]*a
            a=a*10                     #获得list的十进制数字
        num=num+1

        bytes=int(math.log10(num))     #计算数字的位数
        lens_new=bytes+1
        result=[]

        for i in range(lens_new):
            single=int(num/(10**(bytes-i))) #得到每一位的数字
            result.append(single)
            num=num-single*10**(bytes-i)    #将得到的位数减去
        return result

if __name__=="__main__":
    s=Solution()
    a=[9,9,9]
    print(s.plusOne(a))

然后举出比较正常的方法

思路：对每个值遍历判断，如果为9则变为0，并且继续遍历，如果为0到8则加1操作，并且break退出判断

上代码（用时44ms)击败98%

class Solution:
    def plusOne(self,digits):
        """
        :param digits:List[int]
        :return:List[int]
        """
        #
        for i in range(1,len(digits)+1): #对每一个值遍历
            if(digits[-i]==9):   #如果为9则变为0
                digits[-i]=0
            else:                #如果加1则退出
                digits[-i]+=1
                break
        if(digits[0]==0):        #如果全都是9那么第一位为1后面加一个0
            digits[0]=1          #eg:[9,9,9] --> [1,0,0,0]
            digits.append(0)
        return digits
if __name__=="__main__":
    s=Solution()
    a=[5,9,8]
    print(s.plusOne(a))