题目:
函数原型:
int* plusOne(int* digits, int digitsSize, int* returnSize)
注:digits是输入的数组,digitsSize是输入数组的大小,returnSize是输出数组的大小
解析:
先设置数组下标index,初始时指向数组最后一个元素
根据要求,最后一个元素+1
如果最后一个元素+1变为10后,就要进1,即前一个元素+1,。由此设置循环,循环条件为数组元素不再进1且下标index > 0
循环结束后,判断数组第一个元素是否 >= 10,若是则说明输入数组为n个9,最终新数组长度增加1,需要创建新数组返回;若不是则说明数组长度不变,直接返回原数组
关键:为什么设置循环条件数组下标index>0?
元素进1后其前一个元素+1,需要用下标 index - 1,若index=0,则数组向前越界
- int* plusOne(int* digits, int digitsSize, int* returnSize) {
- int index = digitsSize - 1;//设置下标,初始指向最后一个元素
-
- digits[index] += 1;//最后一个元素先加一
-
- while (index > 0 && digits[index] >= 10)//如果元素>=10 直到不再进位1
- {
- digits[index] %= 10;//则元素等于0
- digits[index - 1] += 1;//且前一位进1
- index--;//下标前移
- }
-
- if (digits[0] >= 10)//当最高位也要进位时,说明元素个数增加了,要创建新的数组
- {
- int* returndigits = (int*)malloc(sizeof(int) * (digitsSize + 1));//创建新数组
- memset(returndigits, 0, sizeof(int) * (digitsSize + 1));
- returndigits[0] = 1;//最高位是1
- int i = 1;//从第二位开始赋值
- while (i < digitsSize + 1)
- {
- returndigits[i] = 0;//赋值的都是0
- i++;
- }
- *returnSize = digitsSize + 1;//新数组大小是原数组+1
- return returndigits;//返回新数组
- }
- else//元素个数不变
- {
- *returnSize = digitsSize;//数组大小仍然等于原数组大小
- return digits;//直接返回原数组
- }
- }