题目要求
给定一个非空的数字数组,该数组表示一个非负整数,要求对这个非负整数加上1。返回操作后的数组。
示例
输入:[1,2,3] 表示123,
输出[1,2,4] (123+1=124)
解题思路
因为题目已经说明不考虑负数的情况,所以对于任意非负数进行加1操作,可以分为两种情况:
(1)不需要额外操作,直接加1, 如0,1,2,3…8这些数字。加1后,break返回即可。
(2)需要进位的,如9。那么要把当前位设成0。
如果进位后的下一位也是9怎么办? 所以我们要从后向前对所有数字进行遍历,是9就置0,不是9的话就回到(1)加1然后break。
(eg: 9,19,99 ------> 0,20,00)
最后还要判断一下开头是不是0,如果是0,那么要插入1。
(eg:上述中9,99进行加1后的结果0,00------> 变成10,100)
主要代码c++
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
for(int i=digits.size()-1; i >= 0; i--)
{
if (digits[i] != 9)
{
digits[i] ++;
break;
}
digits[i] = 0;
}
if (digits[0] == 0)
digits.insert(digits.begin(),1);
return digits;
}
};