LeetCode-简单-66-加一(C)

题目

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

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

你可以假设除了整数 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]
 

提示：

1 <= digits.length <= 100
0 <= digits[i] <= 9

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

思路

• 很简单的一题，只要处理好进位就好了。
• 定义plus为进位，初始化为1，以及*returnSize初始化为digitsSize，并分配返回数组returnDigits大小为digitsSize+1。
• 然后开始一个循环：
  • 从digits的尾部(下标为digitsSize-1)开始计算每位与plus的和，并得出其个位与十位。
  • 将plus置为上面得出的十位数，将returnDigits的第digitsSize置为个位数。
• 结束循环后，判断plus==1，
  • 成立则将returnSize置digitsSize+1，将下边0的数置1，表示有进位，并返回returnDigits。
  • 否则返回returnDigts+1。
    • 这里的意思是从数组的第二个值开始返回。

代码

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