【Leetcode算法题】66. 加一

题目描述

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一，最高位数字存放在数组的首位， 数组中每个元素只存储单个数字，你可以假设除了整数 0 之外，这个整数不会以零开头。

示例

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

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

解题思路

class Solution {
    public int[] plusOne(int[] digits) {
        //获取个位数，即数组最后一个元素
        int length=digits.length;
        int lastIndex=length-1;
        //加1
        digits[lastIndex]+=1;
        //判断最后一个元素是否大于9
        if(digits[lastIndex]>9){
            while(lastIndex>0 && digits[lastIndex]>9){
                digits[lastIndex]=0;
                digits[lastIndex-1] += 1;
                lastIndex--;
            }
            //判断是否只有个位数
            if(lastIndex==0 && digits[lastIndex]>9){
                //数组扩容
                digits=new int[length+1];
                digits[0]=1;
            }
        }
        return digits;
    }
}