加一

題目

給定一個由整數組成的非空數組所表示的非負整數，在該數的基礎上加一。

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

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

示例 1:

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

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

鏈接：https://leetcode-cn.com/problems/plus-one

思路

在數組上做加1操作，判斷末尾是否爲9，如果不是，則直接加一，輸出；否則，該位爲0，再判斷。如果數組全爲9，則在數組頭部加一位1

代碼

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        for i in range(1,len(digits)+1):
            if digits[-i]!=9:
                digits[-i]+=1
                return digits
            digits[-i]=0
        #若數組全爲9的操作
        digits.insert(0,1)
        return digits