LeetCode 66.加一
程序员文章站
2024-02-17 10:32:52
...
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储一个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
示例 1:
输入: [1,2,3] 输出: [1,2,4] 解释: 输入数组表示数字 123。
示例 2:
输入: [4,3,2,1] 输出: [4,3,2,2] 解释: 输入数组表示数字 4321。
思路:
观察如下情况加1会发生什么
1.9+1=10 由一位数变为两位数
2.199+1=200 所有的9变为0,9之前的一位加1
3.12345999+1=123456000 所有的9变为0,9之前的一位加1
针对2.3两种情况,由于加1产生进位,可以从后向前遍历整个数组,如果是9则变为0,直到遇到一个不是9的数字,将该数字加1,返回数组。
针对1这种情况,由于加1数位增加了,数组长度必须加1,创建一个新数组,把第一位设置为1,其后digits.length长的数组元素全部设置为0。
代码:
class Solution {
public int[] plusOne(int[] digits) {
int i = digits.length-1;
while(digits[i]==9){
digits[i--]=0;
if(i<0)break;
}
if(i>=0){ //i大于等于0说明没有发生数字位数增加的情况
digits[i]=digits[i]+1;
}
else{ //发生了数字位数增加的情况,创建一个新数组以容纳新的数字
int []newDigits=new int[digits.length+1];
newDigits[0]=1;
for(int j = 1;j<newDigits.length;j++){
newDigits[j]=0;
}
return newDigits;
}
return digits;
}
}