加一
给定一个非负整数组成的非空数组,在该数的基础上加一,返回一个新的数组。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
分析
这道题其实很简单,只存在 进位问题,那么我们考虑一下,进位都存在哪些情况:
- 只末尾进位,如19
- 末尾+中间连续进位,如199
- 末尾+中间+首位进位,如999
对于 1 而言,我们只需要令9->0,1->2即可
对于 2 而言,我们需要令9->0,9->0,1->2
对于 3 而言,物品们需要令9->0,9->0,9->0并扩充数组,首位置 1
因此我们只需要从后向前遍历,逢9置0,非9加1,最后考虑是否扩充数组即可
代码
class Solution {
public int[] plusOne(int[] digits) {
int n = digits.length;
for(int i=n-1;i>=0;i--){
//非9加1
if(digits[i]<9){
digits[i]++;
return digits;
}
//逢9置0
digits[i]=0;
}
//全部为9,则需要数组扩充1位
int[] result = new int[n+1];
result[0] = 1;
return result;
}
}