题源:https://leetcode-cn.com/leetbook/read/top-interview-questions-easy/x2cv1c/
解析:分情况讨论。(1)先讨论全部为9的情况,大数组容量至原来的大小+1,首位为1,其他位为零;(2)不全部为9,从最低位开始遍历,开始将最低位设置为当前位,判断当前位是否为9,不为9直接+1结束,若为9此位为0,进一,将当前位的前一位设置为当位即i--,进行循环。
代码:
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int lenth=digits.size();
int k=0;
for(auto x:digits){
if(x==9) k++;
}
if(k==lenth){///全部是9的情况
digits.resize(lenth+1);///此处将给定数组的容量增加1
for(int i=0;i<=lenth;i++){
if(i==0) digits[i]=1;
else{
digits[i]=0;
}
}
lenth+=1;
}
else{///不全为9的情况
int add=0;
for(int i=lenth-1;i>=0;i--){///从最低位开始遍历
if(digits[i]==9&&add==0){///最低位为9,进一位add=1
digits[i]=0;
add=1;
}
else if(digits[i]==9&&add==1){///此时add==1,证明已经加过1,只处理进位的1即可
digits[i]=0;
add=1;
}
else if(digits[i]<9){///只要当前位不为9,无论是进位还是刚开始的最低位,都需要加1结束
digits[i]+=1;
break;
}
}
}
return vector(begin(digits),begin(digits)+lenth);
}
};