题目描述:
给定一个由 整数 组成的 非空 数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 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]
解答:
class Solution {
public int[] plusOne(int[] digits) {
int n = digits.length;
// 从最后一位开始加1
for (int i = n - 1; i >= 0; i--) {
if (digits[i] < 9) {
// 如果当前位不是9,直接加1并返回
digits[i]++;
return digits;
} else {
// 如果当前位是9,变为0,继续往前加1
digits[i] = 0;
}
}
// 如果所有位都是9,需要在数组前面加一位1
int[] res = new int[n + 1];
res[0] = 1;
return res;
}
}
笔记:
从最后一位开始遍历;
(1)判断非9的情况,直接加1并返回
(2)为9 -> 0,前一位加一
最后动态创建一个新的数组,大小为n+1,即原数字长度+1(防止999,99之类的数字)
只有在都为9的情况下for循环才会完全执行完,res[0]=1
否则
return digits;直接返回输出了。