leetcode题解66-加一

问题描述

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

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

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

示例 1：
输入：digits = [1,2,3]
输出：[1,2,4]
解释：输入数组表示数字 123。
示例 2：
输入：digits= [4,3,2,1]
输出：[4,3,2,2]
解释：输入数组表示数字 4321。
示例 3： 输入：digits = [0]
输出：[1]

提示：
1 <= digits.length <= 100
0 <= digits[i] <= 9

解题思路

只需要数组最低位开始，加上一个进位1，如果遇到某一位变为10，那么就需要继续产生1个进位，在下一位加1，如此直到碰到一个使得某一位的值小于10。如果数组的每一位都产生1个进位，比如9999999等，此时我们需要开辟一个新的数组，长度+1，然后让数组的第一个元素为1，其余皆为0即可

代码实现

class Solution {
    public int[] plusOne(int[] digits) {
        //如果数组中只有一个元素，并且其值为0，就可以直接在0的基础上加1即可
        if(digits[0]==0 && digits.length==1){
            digits[0]=1;
            return digits;
        }
        //我们需要定义一个进位标志位来标志进位情况，因为我们执行的是+1的操作，所以进位初值为1
        int jinwei=1;
        //从数组最后一个元素开始计算
        for(int i=digits.length-1;i>=0;i--){
            int temp=digits[i]+jinwei;
            //如果数组的值加上进位等于10，那么就需要再产生一个进位
            if(temp==10){
                digits[i]=0;
                jinwei=1;
            }else{
            //如果没有产生10，后面就不需要计算了
                digits[i]=temp;
                jinwei=0;
                break;
            }
        }
        //如果因为数组中的值全部是9，加1之后，产生进位，数组长度发生变化，因此前面的循环无法解决此类问题
        if(jinwei==1){
            int [] nums=new int[digits.length+1];
            nums[0]=1;
            for(int i=1;i<nums.length;i++){
                nums[i]=digits[i-1];
            }
            return nums;
        }
        return digits;
    }
}