Leetcode66. 加一
题目:
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例:
输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。
题解:
根据题意加一,没错就是加一这很重要,因为它是只加一的所以有可能的情况就只有两种:
- 除 9 之外的数字加一;
- 数字 9。
加一得十进一位个位数为 0 加法运算如不出现进位就运算结束了且进位只会是一。
所以只需要判断有没有进位并模拟出它的进位方式,如十位数加 1 个位数置为 0,如此循环直到判断没有再进位就退出循环返回结果。
java代码:
public static int[] plusOne(int[] nums) {
int len = nums.length;
for (int i = len - 1; i >= 0; i--) {
nums[i]++;
nums[i] = nums[i] % 10;
if (nums[i] != 0) return nums;
}
nums = new int[nums.length + 1];
nums[0] = 1;
return nums;
}
scala代码如下:
def plusOne(digits: Array[Int]): Array[Int] = {
var digit = digits
val len = digit.length
var i = len - 1
digit(i) += 1
if (digit(i) > 9) {
while (i > 0 && digit(i) > 9) {
digit(i) = 0
i = i - 1
digit(i) += 1
}
if (i == 0 && digit(i) > 9) {
digit = new Array[Int](len + 1)
digit(0) = 1
}
}
digit
}