非空数组表示的元素值加一

解题思路

标签：数组遍历
这道题需要整理出来有三种情况，在进行处理会更舒服

① 末位无进位，则末位加一即可，因为末位无进位，前面也不可能产生进位，比如 45 => 46
② 末位有进位，在中间位置进位停止，则需要找到进位的典型标志，即为当前位 %10后为 0，则前一位加 1，直到不为 0 为止，比如 499 => 500
③ 末位有进位，并且一直进位到最前方导致结果多出一位，对于这种情况，需要在第 2 种情况遍历结束的基础上，进行单独处理，比如 999 => 1000
时间复杂度：O(n)

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */
int* plusOne(int* digits, int digitsSize, int* returnSize){
//*digits原数组， *returnSize输出数组，digitsSize原数组长度

    for(int i = digitsSize - 1; i >= 0; --i){
        digits[i] = digits[i] + 1;//最后元素+1判断是不是10
        //如果当前元素不为10，直接返回数组
        if(digits[i] != 10){
            *returnSize = digitsSize;
            return digits;
        }
        //当第一个元素不为10，后面元素均为10的情况
        if(digits[i] == 10)
            digits[i] = 0;
    }
    
    //元素全为9开辟新数组
    int* ans = malloc(sizeof(int) * (digitsSize + 1));
    memset(ans, 0, sizeof(int) * (digitsSize + 1));//将*ans数组元素全部置0
    ans[0] = 1;//第一个元素为1
    *returnSize = digitsSize + 1;//*returnSize长度为 digitsSize + 1
    return ans;//返回新数组
}