leetcode:66. 加一

题目:

leetcode:66. 加一_第1张图片

函数原型:

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

你可能感兴趣的:(leetcode刷题训练营,leetcode,算法,职场和发展)