加一
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储 单个 数字。
你可以假设除了整数 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
思路
逐个计算
TypeScript
function plusOne(digits: number[]): number[] {
const num = digits.pop() + 1
digits.push(num);
if (num === 10) {
for (let i = digits.length - 1; i >= 0; i--) {
if (digits[i] === 10) {
digits[i] = 0;
if (i - 1 === -1) digits.unshift(1);
else digits[i - 1]++
}
}
}
return digits;
};
找出最长的后缀 9
从 1 阶开始记录所有阶数有几种方法, n 阶 = (n-1)的方法数量 + (n-2)的方法数量
TypeScript
function plusOne(digits: number[]): number[] {
for (let i = digits.length - 1; i >= 0; --i) {
if (digits[i] !== 9) {
digits[i]++;
for (let j = i + 1; j < digits.length; ++j) {
digits[j] = 0;
}
return digits;
}
}
// digits 中所有的元素均为 9
const arr = new Array(digits.length + 1).fill(0);
arr[0] = 1;
return arr;
};
优化
最后一位++后等于 10 则往前处理
TypeScript
function plusOne(digits: number[]): number[] {
for (let i = digits.length - 1; i >= 0; i--) {
digits[i] = (digits[i] + 1) % 10;
if (digits[i] !== 0) {
return digits;
}
}
// digits 中所有的元素均为 9
const arr = new Array(digits.length + 1).fill(0);
arr[0] = 1;
return arr;
};