leetcode 66. 加一

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

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

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

示例 1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。

示例 2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/plus-one
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* plusOne(int* digits, int digitsSize, int* returnSize){
    int start = 0;
    int i = digitsSize - 1;
    digits[i] += 1;
    while (digits[i] == 10 && i > 0) {
        digits[i] = digits[i]%10;
        i--;
        digits[i] += 1;
    }

    *returnSize = digitsSize;
    if (digits[0] == 10) {
        digits[0] = digits[0] % 10;
        *returnSize += 1;
    }

    int *ans = malloc(sizeof(int) * (*returnSize));
    if (*returnSize != digitsSize) {
        ans[0] = 1;
        start = 1;
    }

    for (i = 0; start < *returnSize; i++, start++) {
        ans[start] = digits[i];
    }

    return ans;
}

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* plusOne(int* digits, int digitsSize, int* returnSize){
    int start = 0;
    *returnSize = digitsSize;
    for (int i = digitsSize - 1; i >= 0; i--) {
        if (digits[i] != 9) {
            digits[i] += 1;
            return digits;
        }
        else {
            digits[i] = 0;
        }
    }

    *returnSize += 1;
    int *ans = malloc(sizeof(int) * *returnSize);
    ans[0] = 1;
    for (int i = 0; i < digitsSize; i++) {
        ans[i + 1] = digits[i];
    }

    return ans;
}

class Solution:
    def plusOne(self, digits: List[int]) -> List[int]:
        for i in range(len(digits)-1,-1,-1):
            if digits[i] != 9:
                digits[i] += 1
                return digits
            else:
                digits[i] = 0

        digits.insert(0,1)
        return digits