LeetCode——66. 加一

题目

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。

最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。

你可以假设除了整数 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;
        }
    }
}

结果

一道简单题，不过太久没做这些，思维开始钝化了！