题源力扣:
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
假设除了整数 0 之外,这个整数不会以零开头。
C++:
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
//从后往前找到第一个不为9的位置j
int j=-1;
for(int i=digits.size()-1;i>-1;){
if(digits[i]==9)
i--;
else{
j=i;
break;
}
}
//不全为9,j位置+1,j后全置0
if(j!=-1){
digits[j]=digits[j]+1;
for(int k=j+1;k<digits.size();k++)
digits[k]=0;
return digits;
}
//原数组全为9.则新size+1,首=1其余0
else{
vector<int> res(digits.size()+1,0);
res[0]=1;
return res;
}
}
};