题目来源:每日一练第4天:加一
同时也是:LeetCode 66.加一
题目详情
给定一个由整数
组成的非空
数组所表示的非负整数,在该数的基础上加一。最高位数字存放在数组的首位, 数组中每个元素只存储单个
数字。你可以假设除了整数 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
解题思路
这道题目的意思就是给定一个非负的整数,让你把它加一,然后再输出,因为是把整数转成一个整数数组给我们的,所以我们需要做的是逆序遍历数组,把每一位数组元素加一,如果该元素加1正好等于10了,那么就需要考虑进位,将原数组元素变为0,加进位得到的1带入下一个数组元素,直到遍历结束。
当然有一种特殊情况需要考虑,例如digits为[9,9,9,9],此时每位数组元素都是需要进位的,此时digits变成了[0,0,0,0],最高位的1却因为数组长度受限,没办法表示出来,我们就需要创建一个长度为digits数组长度加一的数组,再将数组的第一位赋值为1,然后返回该数组即可。
代码实现
public class PlusOne {
public static void main(String[] args) {
Solution solution = new PlusOne().new Solution();
// to test
int[] digits = {9,9,9,9};
int[] result = solution.plusOne(digits);
for(int i = 0 ; i < result.length ; i++) {
System.out.print(result[i]);
}
}
class Solution {
public int[] plusOne(int[] digits) {
int len = digits.length - 1;
for(int i = len ; i >= 0 ; i--) {
int num = digits[i] + 1;
if(num == 10) {
digits[i] = 0;
} else {
digits[i] = num;
break;
}
}
if(digits[0] == 0) { // 如果出现最高位要进位,则拓展数组
int result[] = new int[digits.length+1];
result[0] = 1;
return result;
}
return digits;
}
}
}