LeetCode66:加一

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

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

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

示例 1:

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

示例 2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

解析:

        该题目主要考虑加一之后的进位问题,所以需要设置一个变量保存是否有进位信息。如果最后一位没有进位信息,则整个数组都不可能有进位,所以,此时直接返回原数组最后一位加一就可以了;如果最后一位出现了进位,则循环查看其他位是否有进位,如果有进位,则继续循环,否则将不再有进位,直接返回就可以了。如果到最高位有进位,则需要在数组最前面补一个1,可以使用vector的insert方法。

代码:

        

vector<int> plusOne(vector<int>& digits)
{
	int size = digits.size();//数组大小
	int sum = digits[size - 1] + 1;
	if (sum < 10)//最后一位是否有进位
	{
		digits[size - 1] = sum;
		return digits;
	}
	int carry = 1;//进位的标志
	digits[size - 1] = 0;
	for (int i = size - 2; i >= 0; i--)
	{
		if (digits[i] + carry >= 10)//如果有进位carry设为1,原值为0
		{
			digits[i] = 0;
			carry = 1;
		}
		else//没有进位,则原值加carry,carry设为0,返回数组
		{
			digits[i] = digits[i] + carry;
			carry = 0;
			return digits;
		}
	}
	if (carry == 1)//最高位还有进位,则补1
	{
		digits.insert(digits.begin(), 1);
	}
	return digits;
}

猜你喜欢

转载自blog.csdn.net/qq_36214481/article/details/84999095