【leetcode 簡單】第十五題 加一

給定一個非負整數組成的非空數組，在該數的基礎上加一，返回一個新的數組。

最高位數字存放在數組的首位， 數組中每個元素只存儲一個數字。

你可以假設除了整數 0 之外，這個整數不會以零開頭。

示例 1:

輸入: [1,2,3]
輸出: [1,2,4]
解釋: 輸入數組表示數字 123。

示例 2:

輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入數組表示數字 4321。

class Solution:
    def plusOne(self, digits):
        """
        :type digits: List[int]
        :rtype: List[int]
        """
        last = len(digits) - 1
        for i in range(len(digits)):
            if digits[last] == 9:
                digits[last] = 0
                last -= 1
            else :
                digits[last] = digits[last]+1
                break
        if max(digits) == 0:
            digits.insert(0,1)
        return digits