【leetcode C语言实现】66.加一

题目描述

给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。

最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1：

输入：digits = [1,2,3]
输出：[1,2,4]
解释：输入数组表示数字 123。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/plus-one

解题思路

本题重点在于进位的情况，因此可以分成以下几种：
（1）个位上都没有9，则直接将个位加1即可
（2）个位为9或最低几位都为9，但最高位不为9
（3）所有位都为9
由于该数字由数组表示，因此需要逐个遍历数组中的元素，且应从个位也就是数组的最右边开始遍历，若遍历到的位不为9，则直接将该位加1并终止循环；若遍历到的位为9，则将9+1并对10取余得0，此时将加1进位到下一位。若遍历到最高位也就是最左边一位还没有终止循环，则表示该数字所有位都为9，此时需要将该数字位数加1，对应于该题由于原数组大小固定，只能重新申请新数组的内存，并将array[0]置为1表示该数字最高位为1。

代码

int* plusOne(int* digits, int digitSize, int* returnSize)
{
	int i;

	for (i = digitSize - 1; i >= 0; i--)
	{
		if (digits[i] == 9)
		{
			digits[i] = 0;
		}
		else
		{
			digits[i]++;
			*returnSize = digitSize;
			return digits;
		}
	}

	int* new_array = (int *)calloc((digitSize + 1), sizeof(int));
	new_array[0] = 1;
	*returnSize = digitSize + 1;

	return new_array;
}

测试代码及结果

进行测试时需要综合考虑该函数功能测试及边界值测试，功能测试包括数字没有进位的情况、数字中间有进位的情况及数字所有位都进位的情况，边界值测试包括输入空指针等。

void print_digits(int* array, int size)
{
	int i;

	if (array == NULL || size == 0)
		return;

	for (i = 0; i < size; i++)
	{
		printf("%d ", array[i]);
	}
	printf("\n");
}

int main()
{
	// ======功能测试======
	// 1. 没有进位
	int digits1[] = { 1, 2, 3 };
	int len1 = sizeof(digits1) / sizeof(digits1[0]);
	int new_size1 = 0;
	int *new_ptr1;

	printf("没有进位：\n");
	new_ptr1 = plusOne(digits1, len1, &new_size1);
	print_digits(new_ptr1, new_size1);


	// 2. 中间有进位
	int digits2[] = { 1, 9, 3 };
	int len2 = sizeof(digits2) / sizeof(digits2[0]);
	int new_size2 = 0;
	int* new_ptr2;

	printf("中间有进位：\n");
	new_ptr2 = plusOne(digits2, len2, &new_size2);
	print_digits(new_ptr2, new_size2);


	// 3. 全是进位
	int digits3[] = { 9, 9, 9 };
	int len3 = sizeof(digits3) / sizeof(digits3[0]);
	int new_size3 = 0;
	int* new_ptr3;

	printf("全是进位：\n");
	new_ptr3 = plusOne(digits3, len3, &new_size3);
	print_digits(new_ptr3, new_size3);
	
	// ======特殊输入测试======
	int* digits4 = NULL;
	int new_size4 = 0;
	int* new_ptr4;

	printf("特殊输入：\n");
	new_ptr4 = plusOne(digits4, 0, &new_size3);
	print_digits(new_ptr4, new_size4);

	return 0;
}

执行结果

时间复杂度：由于需要遍历数组一遍，因此时间复杂度为O(n)。
空间复杂度：最差情况下该数组各个元素都为9，此时需要重新分配数组内存，因此空间复杂度为O(n)。