题目描述:
给定一个非负数,表示一个数字数组,在该数的基础上+1,返回一个新的数组。
该数字按照大小进行排列,最大的数在列表的最前面。
样例:
给定 [1,2,3] 表示 123, 返回 [1,2,4].
给定 [9,9,9] 表示 999, 返回 [1,0,0,0].
思路:看到这道题我们都能反应出来是加法的进位问题。OK,那么很明显就是从最后一位开始加。那么就会分两种情况,一种是9,一种是小于9。很显然如果小于9,那么直接+1return就完事;如果大于9,就将这位置0继续加,如果上一位是小于9的,+1return,否则再次重复这个过程。将这个问题的本质想清楚就会很容易写代码了。
class Solution {
public:
/*
* @param digits: a number represented as an array of digits
* @return: the result
*/
vector<int> plusOne(vector<int> &digits) {
// write your code here
vector<int> res;
if (digits.size() == 0) return res;
int size = digits.size() - 1;
for (int i = size; i >= 0; i--)
{
if (digits[i] != 9)
{
digits[i] += 1;
return digits;
}
else
{
digits[i] = 0;
if (i == 0)
{
digits[i] = 1;
digits.push_back(0);
}
}
}
return digits;
}
};