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