给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3] 输出: [1,2,4] 解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1] 输出: [4,3,2,2] 解释: 输入数组表示数字 4321。
我对这道题的思路很简单,就是对最后一位加一然后输出,对于一种需要进位的情况,就需要用到vector这个STL的用法,那就是用push_back来在最后一位加0,并且把除了第一位的所有位变为0,然后将第一位置1,但是空间消耗较多,因为是逐一遍历的,下面给出AC代码:
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int len=digits.size();
if(len==1&&digits[0]!=9){
digits[0]+=1;
return digits;
}
if(len==1&&digits[0]==9){
digits[0]=1;
digits.push_back(0);
return digits;
}
else{
digits[len-1]+=1;
for(int i=len-1;i>=0;i--){
if(i==0)
break;
else{
if(digits[i]==10&&i!=0){
digits[i]=digits[i]%10;
digits[i-1]+=1;
if(digits[i-1]==10&&i-1==0){
for(int i=len-1;i>0;i--){
digits[i]=0;
}
digits[0]=1;
digits.push_back(0);
}
}
}
}
}
return digits;
}
};