LeetCode 66 加一 Java用时 1ms

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

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

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。

示例 2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

 

思路：

1. 如果最后一位不是9，则直接最后一位+1返回
2. 如果最后一位是9，则最后一位改为0，判断倒数第二位是不是9，做相同的操作，一旦遇到某一位不是9就直接返回。需要注意的是，若所有数组全都是9，则需要我们把所有位改为0，此时再在前面加一个1。

Java代码如下：

class Solution {
    public int[] plusOne(int[] digits) {
       int len=digits.length;
       int digitsB[]=new int[len+1];      //定义长度为len+1的数组 存放全为9的情况             
       if(digits[len-1]<9)  //处理数组最后一位不为9的情况，正常加1
        {
          digits[len-1]=digits[len-1]+1;
          return digits;
        } 
        else 
        {
            int l=len-1;
            while(l>=0&&digits[l]==9)  //可以处理数组只有一位数，且其为9的情况
                l--;
            if(l<0)  //数组全为9的情况 需要用 digitsB数组，扩充一位，其他情况无需使用。
            {
               for(int i=1;i<len+1;i++)
                 digitsB[i]=0;
               digitsB[0]=1;
               return  digitsB;
            }               
            digits[l]=digits[l]+1;   //找到数组中数字不为9的下标，对该数加1
            for(int i=l+1;i<len;i++)
                digits[i]=0;
        }          
        return digits;
    }
}

附上AC成功图：