给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入:digits = [1,2,3] 输出:[1,2,4] 解释:输入数组表示数字 123。
示例 2:
输入:digits = [4,3,2,1] 输出:[4,3,2,2] 解释:输入数组表示数字 4321。
示例 3:
输入:digits = [0] 输出:[1]
提示:
1 <= digits.length <= 100
0 <= digits[i] <= 9
参考思路:
从后往前遍历数组,如果最后一位加一小于10,则直接结束,返回数组;否则,把最后一位设置为0,继续往前遍历加一。
这时候提交忘记考虑加一后数组变大一位的情况,即9、99、999变为10、100、1000等情况,这时候我们单独来进行考虑:
当然这种情况也是在之前操作之后,遍历整个数组,如果全为0,则证明是我们上述提到的情况,这时候就需要创建一个新的数组,数组大小加一,将数组第一个值设置为1,其他不用理会即可,或者后面设置为0,这种情况单独返回新数组,其余情况返回原数组。
代码如下:
class Solution {
public:
vector<int> plusOne(vector<int>& digits) {
int flag=0;
for(int i=digits.size()-1;i>=0;i--){
if((digits[i]+1)<10){
digits[i]++;
break;
}
else{
digits[i]=0;
}
}
for(int i=0;i<digits.size();i++)
{
if(digits[i]!=0){
flag=1;
break;
}
}
if(flag==0){
vector<int> digit(digits.size()+1);
digit[0]=1;
digit[digit.size()-1]=0;
return digit;
}
return digits;
}
};
可以在评论区交流下思路哦!