Leetcode 66 加一

题目

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

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

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

解题思路

  先进位，再考虑溢出的情况。

代码

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