题目描述:
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
解题思路:
首先,当最后一位元素不是9,直接加一返回即可;当最后一位元素是9时,考虑逐级判断进位。
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int len=digits.size();
if(digits[len-1]!=9){
digits[len-1]+=1;
return digits;
}
else{
int add=1;
digits[len-1]=0;
for(int i=len-2;i>=0;i--){
if(add==1){
digits[i]+=1;
if(digits[i]%10==0){
digits[i]=0;
add=1;
}
else{
add=0;
break;
}
}
}
if(add==1){
digits.resize(len+1);
digits[0]=1;
}
}
return digits;
}
};