算法练习(15) leetcode 66. 加一

https://leetcode-cn.com/problems/plus-one/

class Solution {
    public int[] plusOne(int[] digits) {
        boolean flag=false;
        digits[digits.length-1]=digits[digits.length-1]+1;
        for (int i=digits.length-1;i>=0;i--){
            if (digits[i]>9){
                if (i==0) {
                    flag=true;
                    digits[i]=digits[i]%10;
                } else {
                    digits[i-1]=digits[i-1]+1;
                    digits[i]=digits[i]%10;
                }
            }
        }
        if (flag) {
            int[] ans=new int[digits.length+1];
            ans[0]=1;
            for (int i=0;i<digits.length;i++){
                ans[i+1]=digits[i];
            }
            return ans;
        }else {
            return digits;
        }
    }
}