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;
}

You may also like...

发表评论

您的电子邮箱地址不会被公开。 必填项已用*标注