leetcode66--加一

题目描述

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

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

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

示例 1:

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

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

题解

这道题看到以后，我首先想到的是分情况讨论，一种情况是类似9999这种，加1以后数组会边长，另外一种情况就是加1以后，数组不会加长的。因此代码安装两种情况来分别解决即可。对于第二种情况，只需要对数组从后向前进行遍历，遇到9则变0，遇到非9则加1并跳出循环即可。

class Solution {
    public int[] plusOne(int[] digits) {
    //用于判断数组是否是第一种情况的
        int flag=1;
        for(int i=0;i<digits.length;i++){
            if(digits[i]!=9){
                flag =0;
                break;
            }
        }
        int[] new_digits = new int[digits.length+1];
        if(flag==1){
            new_digits[0]=1;
            return new_digits;
        }else{
            for(int i=digits.length-1;i>=0;i--){
                if(digits[i]==9){
                    digits[i]=0;
                }else{
                    digits[i]+=1;
                    break;
                }
            }
        }
        return digits;

        

    }
}

用时0ms。