leetcode题目66. 加一

题目描述

链接：https://leetcode-cn.com/problems/plus-one/
给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。

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

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

示例

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

输入：digits = [1,2,9]
输出：[1,3,0]
解释：输入数组表示数字 129。

代码

   public int[] plusOne(int[] digits) {
        int flag = 0;
        for(int i = digits.length - 1; i >= 0; i--) {

            int sum = digits[i] + flag;
            if (i == digits.length - 1) {
                sum = sum + 1;
            }
            digits[i] = sum % 10;
            flag = sum / 10;
        }
        if (flag <= 0) {
            return digits; 
        }

        // digits 中所有的元素均为 9
        int[] ans = new int[digits.length + 1];
        ans[0] = 1;

        return ans;
    }