Leetcode 题解 [66.加一](C语言)

Leetcode 题解（每日打卡）

[66.加一]

题目链接

题目描述

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

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

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

思路

将数组从后往前遍历，若取出的元素为9的话，则令该位置的值为0，继续遍历，若当i>0时，该值不为9，则该值加1，并返回数组，否则创建新数组，并令数组第0个元素为1，其余元素均为0。

其实该题总共就只有三种情况:

1. 不进位
2. 进位，但是进位后，数组的总长度不变
3. 进位，进位后，数组的总长度加1（全为9的情况）

代码实现

int* plusOne(int* digits, int digitsSize, int* returnSize){
    for(int i = digitsSize-1;i >= 0;i--){
        if(digits[i] == 9){
            digits[i] = 0;
            if(i == 0){
               *returnSize = digitsSize+1;
            }   
        }else{
            digits[i] = digits[i]+1;
            *returnSize = digitsSize;
            return digits;
        }
    }
    int* arr = (int*)malloc(sizeof(int)*(*returnSize));
    arr[0] = 1;
    for(int i = 1;i < *returnSize;i++){
        *(arr+i) = 0;
    }
    return arr;
}