用C语言实现数组值加一问题

1.题目描述  

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

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

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

2.解题方法

        找出最长的后缀9

3.解题思路

首先确定加一是在数组digits最后一位加上的，那么判断进位时可以倒着遍历。可以分为三种情况：

· 最后一位不为9，比如[1, 2, 3], 那么直接将最后一位加一，并返回 [1, 2, 4]；

· 倒着遍历，其中有若干个9，比如[1, 2, 9, 9] ，那么应找到 倒着的最后一个 9，然后令它前一项，也就是2加一，其余9全部置0, 返回 [1, 3, 0, 0]；

· 如果digits[0]也为9，比如[9 ,9, 9, 9]， 那么应该对digits重新分配比原来多1的内存。digits[0] 置1，其余全部置0，返回[1, 0, 0, 0, 0]。

大家可能并不明白这个returnSize是什么意思，其实它就是数组需要返回的长度。前两种情况下returnSize = digitsSize，第三种情况下returnSize = digitsSize+1。

4.程序代码

int* plusOne(int* digits, int digitsSize, int* returnSize){ //returnSize表示返回数组的长度
    digits[digitsSize - 1] += 1; //先给最后一项加一
    int i = digitsSize - 1; // 从最后一项开始遍历
    while(digits[i] > 9 && i > 0) { //倒着找到最后一个9，然后正着包括这个9往后都置零
             digits[i - 1] += digits[i] / 10;
            digits[i] = 0;
            i--;
    }
    if (digits[0] > 9) { //最后一项出现9，加一后原数组长度加一
        ++digitsSize;
        // 利用realloc对数组digits重新分配内存
        digits = (int *)realloc(digits, (digitsSize) * sizeof(int)); 
        digits[0] /= 10;
        digits[digitsSize - 1] = 0;
        *returnSize =  digitsSize; //returnSize需要返回值
    } else {
        *returnSize =  digitsSize;
    }  
    return digits;
}