函数原型:
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;//直接返回原数组 } }
