数组/数学简单 LeetCode66. 加一

66. 加一

描述

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

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