问题描述:
给定一个非负数,表示一个数字数组,在该数的基础上+1,返回一个新的数组。
该数字按照数位高低进行排列,最高位的数在列表的最前面。
样例:
输入:[1,2,3]
输出:[1,2,4]
输入:[9,9,9]
输出:[1,0,0,0]
思路:
重点是最后一位,只有最后一位是9的情况才涉及到进位。
如果最后一位不是9,直接把最后一位加一再返回即可;
否则从最后一位开始,依次向前考虑进位,直到不进位为止。如果进位,就把当前位置的数置零,前一位加一。
如果遇到99,999,9999这样的情况,按照上述算法流程,会产生全零数组,因此要在数组第零位之前插入1
算法:
vector<int> plusOne(vector<int> &digits) {
int size = digits.size();
if(digits[size-1] != 9){
digits[size-1]++;
return digits;
}
digits[size-1]=0;
for(int i=size-2; i>=0; i--){
digits[i] += 1;
if(digits[i]==10){
digits[i] = 0;
}
else break;
}
if(digits[0]==0){
digits.insert(digits.begin(), 1);
}
return digits;
}