题目
给定一个由整数组成的非空数组所表示的非负整数,在该数的基础上加一。
最高位数字存放在数组的首位, 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外,这个整数不会以零开头。
思路
相当于把数组看成一个整数,加一之后再放回到这个数组里面。我们可以区分一下情况。
什么情况下需要进位?
只要在最后一个数是9的时候,才需要进位,不然的话,我们直接把最后一个数加上1返回即可;
什么情况下需要改变数组?
只有当所有的数组中所有的数字都是9的时候,我们才需要新建一个新的数组,并且第0位是1,其余位置都是0,然后返回即可。
除去这两种情况,就剩下最后一位是9,但是数组中的数字不全为9的情况,只要从右往左开始,第一个不为9的位置给加上1,其余位置变为0即可解答。
代码
class Solution {
public int[] plusOne(int[] digits) {
int length=digits.length;
if(digits[length-1]!=9)
{
++digits[length-1];
return digits;
}
int count=0;//计算9的个数
for (int i=0;i<length;++i)
if (digits[i]==9)
++count;
if (count==length)//整个数组都是9
{
int result[]=new int[length+1];
result[0]=1;
for (int i=1;i<length+1;++i)
result[i]=0;
return result;
}
else //数组中的数不全为9,那么久表明这个数组加1之后可以存放在原来的数组中
{
int index=length-1;
while (index>=0)
{
digits[index]=(digits[index]+1)%10;
--index;
if ((digits[index]+1)%10!=0)
{
digits[index]=digits[index]+1;
break;
}
}
return digits;
}
}
}
结果
一道简单题,不过太久没做这些,思维开始钝化了!