LeetCode解題之Plus One

原題

給一個由包含一串數字的列表組成的非負整數加上一。

注意點：

例子：

輸入: [1, 2, 3, 4, 9]

輸出: [1, 2, 3, 5, 0]

解題思路

從低位到高位，如果後一位有進位的話，那麼該位要加上一，否則退出循環。如果最高位也進位，那麼在列表前要插入一個一。

AC源碼

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


if __name__ == "__main__":
    assert Solution().plusOne([1, 2, 3, 4, 9]) == [1, 2, 3, 5, 0]
    assert Solution().plusOne([9]) == [1, 0]