[Leetcode]@python 66. Plus One

题目链接：https://leetcode.com/problems/plus-one/

 

题目大意：用一个digit数组表示一个非负整数，加1后返回这个数组

　　　　例如：[9,9]加1返回[1,0,0]

解题思路：从后往前遍历，一次判断进位即可；注意计算到最高位的时候是否有进位的处理

 

class Solution(object):
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        length = len(digits)
        if length == 0:
            return [1]
        carry = 0
        digits[length - 1] += 1
        while length > 0:
            digits[length - 1] += carry
            if digits[length - 1] > 9:
                carry = 1
                digits[length - 1] = 0
            else:
                carry = 0
                break
            length -= 1
        if carry == 1:
            digits.insert(0, 1)
        return digits