LintCode Python 简单级题目 407.加一

题目描述：

给定一个非负数，表示一个数字数组，在该数的基础上+1，返回一个新的数组。

该数字按照大小进行排列，最大的数在列表的最前面。

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给定 [1,2,3] 表示 123, 返回 [1,2,4].

给定 [9,9,9] 表示 999, 返回 [1,0,0,0].

标签 

数组 谷歌

 

题目分析：

末尾[-1]加1，然后逆序循环，判断元素是否为10，然后处理；

特殊处理，[0]元素为10，需要在首位插入一个1.

 

源码：

class Solution:
    # @param {int[]} digits a number represented as an array of digits
    # @return {int[]} the result
    def plusOne(self, digits):
        # Write your code here
        n = len(digits)
        digits[-1] += 1
        for i in range(-1,-n,-1):
            if digits[i] == 10:
                digits[i] = 0
                digits[i-1] += 1
        if digits[0] == 10:
            digits[0] = 0
            digits.insert(0,1)
        return digits
    
    def _plusOne(self, digits):
        # Write your code here
        digits.reverse()
        n = len(digits)
        digits[0] += 1
        for i in range(n):
            if digits[i] == 10:
                digits[i] = 0
                if i == n-1:
                    digits.append(1)
                else:
                    digits[i+1] += 1
        if digits[-1] == 10:
            digits[-1] = 0
            digits.append(1)
        digits.reverse()
        return digits