[leetcode66] 加一

【leetcode66】 加一

题目概述

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

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

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

示例 1:

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

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

解决方案

1. 从数组尾开始遍历
2. 如果数字是9则将数字赋值0，并继续向前遍历；直到数字不是9或者遍历到第一个数字
3. 如果不是9，则该数字加一后退出遍历
4. 若加一后的第一个元素是0（证明存在进位），给结果数组添加一个“1”
5. 把其他元素依次加入到数组中

代码

vector<int> plusOne(vector<int>& digits) {
	vector<int> res;
	int i = digits.size() - 1;
	while (i >= 0)
	{
		if (digits[i] == 9)
		{
			digits[i] = 0;
			i--;
			continue;
		}
		else
		{
			digits[i]++;
			break;
		}
	}

	if (digits[0] == 0)
	{
		res.push_back(1);
	}
	for (int j = 0; j < digits.size(); j++)
	{
		res.push_back(digits[j]);
	}

	return res;
}