【力扣】66.加一--Python实现

【题目描述】
给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。

最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1：
输入：digits = [1,2,3]
输出：[1,2,4]
解释：输入数组表示数字 123。

示例 2：
输入：digits = [4,3,2,1]
输出：[4,3,2,2]
解释：输入数组表示数字 4321。

示例 3：
输入：digits = [0]
输出：[1]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/plus-one

【解题思路】
这道题要想明白哪种情况下需要给当前位+1：

• 数组的最后一个元素需要+1
• 如果上一位有进位，则这一位需要+1

把这个+1设成flag，如果需要+1，则flag=1；否则flag为0。当整个循环结束flag还为1时，给整个数组的第一个位置添加一个1。用Python实现的代码如下：

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