数组代表数字加一算法

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

示例 1:

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

示例 2:

输入: [4,3,2,1] 输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

解决方式

class Solution {
    public int[] plusOne(int[] digits) {
         int index = digits.length - 1;

        while (index != -1 && index != isNeedMoveToLeft(index, digits)) {
            digits[index] = 0;
            index--;
        }
        if (index == -1) {
            int[] tempArray = new int[digits.length + 1];
            tempArray[0] = 1;
            for (int i = 0; i < digits.length; i++) {
                tempArray[i + 1] = digits[i];
            }
            return tempArray;
        } else {
            digits[index] = digits[index] + 1;
        }
        return digits;
    }

    public int isNeedMoveToLeft(int index,int[] digits) {
        if(digits[index] == 9){
            return index - 1;
        }
        return index;
    }
}

你可能感兴趣的:(数据结构与算法)