给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
题意:用一个数组来表示一个整数并且加1
初看似乎非常简单,实际上陷阱颇多,如输入9时数组应该返回数组应该是[1,0]因为每一个元素只能代表0~9,这就产生进位的问题了,比较特殊的是类似于999+1,这种形式下输出应该为[1,0,0,0],我的解决办法是,先给原数组最后一位+1,然后对返回数组进行遍历检查有没有为10的,如果有则其前一位应加1,为10处变为0,如果是第一个元素为10,则直接将第一个元素变为1后,在其最后补0即可
代码如下:
```cpp
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
vector<int> res;
digits[digits.size()-1]++;
for(int i=0;i<digits.size();i++){
res.push_back(digits[i]);
}
for(int j=res.size()-1;j>=0;j--){//对返回数组遍历
if(res[j]==10 && j!=0){
res[j]=0;
res[j-1]=res[j-1]+1;
}
if(res[0]==10){//如果是第一位的情况
res[0]=1;
res.push_back(0);
}
}
return res;
}
};
```