加一问题求解

问题描述：

给定一个非负整数组成的非空数组，在该数的基础上加一，返回一个新的数组。

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

可以假设整数不包含任何前导零，除了数字0本身。

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

示例：

输入：[1,2,3]

输出：[1,2,4]

解释：输入数组表示数字123，输出数组表示124。

刚开始的时候，我把输入数组先转换成对应的整数，然后进行加一，再转换成相应的数组，这样做太复杂了。下面是直接对数组中元素判断加一的方法。

vector<int> plusOne(vector<int>& digits) {
	size_t len = digits.size();
	for (int i = len - 1; i >= 0; --i) {
		if (digits[i] != 9) {
			digits[i] += 1;
			return digits;
		}
		else
			digits[i] = 0;
	}
	return digits;
}
int main()
{
	vector<int> v = {1,2,3,4};
	plusOne(v);
	for (auto i : v)
		cout << i << " ";
	cout << endl;
	system("pause");
	return 0;
}