给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
思路
从后往前取变化区域,其实就是求X(不为9)999…的区域,+1后全部置0,X++,digits在这之前的区域不会受影响,如果全为999…,那么前面还要插入1.
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int n=digits.size();
int j=n-1;
bool c=0;
for(int i=n-1;i>=0;--i)
{
if(digits[i]<9)
{
j=i;
break;
}
if(i==0)
j=0;
}
for(int i=n-1;i>=j;--i)
{
if(i!=j)
digits[i]=0;
else
{
if(digits[i]==9)
{
digits[i]=0;
digits.insert(digits.begin(),1);
}
else
digits[i]+=1;
}
}
return digits;
}
};