题解 | #给数组加一#

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param nums int整型一维数组 
     * @return int整型一维数组
     */
    public int[] plusOne (int[] nums) {
        // write code here
        Stack<Integer> stack = new Stack<>();
        int carryBit = 1;
        for (int i = nums.length - 1; i > -1; i--) {
            int tmp = nums[i] + carryBit;
            stack.push(tmp % 10);
            carryBit = tmp / 10;
        }
        if (carryBit == 1) {
            stack.push(1);
        }
        int[] res = new int[stack.size()];
        int index = 0;
        while (!stack.isEmpty()) {
            res[index] = stack.pop();
            index++;
        }
        return res;
    }
}