加一

給定一個由整數組成的非空數組所表示的非負整數，在該數的基礎上加一。

最高位數字存放在數組的首位， 數組中每個元素只存儲一個數字。

你可以假設除了整數 0 之外，這個整數不會以零開頭。

示例 1:

輸入: [1,2,3]
輸出: [1,2,4]
解釋: 輸入數組表示數字 123。
示例 2:

輸入: [4,3,2,1]
輸出: [4,3,2,2]
解釋: 輸入數組表示數字 4321。

評論區代碼：（java)

class Solution {
    public int[] plusOne(int[] digits) {

        Stack<Integer> stack = new Stack<>();
        boolean flag = false;
        digits[digits.length -1] += 1;
        for (int i = digits.length-1; i >= 0; i--) {
            if (digits[i] < 10) {
                return digits;
            } else {
                digits[i] = 0;
                if (i != 0) {
                    digits[i-1] += 1;
                } 

            }
        }
        int[] res = new int[digits.length + 1];
        res[0] = 1;
        System.arraycopy(digits, 0,res, 1, digits.length);
        return res;
    }
}

我的代碼

class Solution {
    public int[] plusOne(int[] digits) {
        int length = digits.length;
        int out[] = new int[length];
        int out2[] = new int[length+1];
        int mark = 0;//進位標誌
        int mark2 = 0;//輸出標誌
        //init out[]
        out2[0] = 1;
        for(int i = 1;i<length;i++){
            out2[i] = 0;
        }
        for(int i = 0;i < length;i++){
            out[i] = digits[i];
            System.out.println(out[i]);
        }
        for(int i = length-1;i>-1;i--){
            if(i==length-1){
                if(digits[i]!=9){
                    out[i]++;
                    break;
                }
                else{
                    mark = 1;//進位
                    out[i] = 0;
                }
            }
            else if(digits[i] !=9 && mark == 1){
                out[i]++;
                break;
            }
            else{
                out[i] = 0;
            }
            if(i == 0 && digits[i] == 9){
                mark2 = 1;
            }
        }
        if(mark2==0){
            return out;
        }
        else{
            return out2;
        }
    }
}

改進後：

class Solution {
    public int[] plusOne(int[] digits) {
        int length = digits.length;
        int out[] = new int[length+1];
        //int mark = 0;//進位標誌
        //int mark2 = 0;//輸出標誌
        out[0] = 1;
        for(int i = 1;i<length+1;i++){
            out[i] = 0;
        }
        digits[length-1]++;
        for(int i = length-1;i>0;i--){
            if(digits[i] < 10){
                return digits;
            }
            else{
                digits[i] = 0;
                digits[i-1]++;
            }
        }
        if(digits[0] == 10){
            return out;
        }
        else{
            return digits;
        }
    }
}

改進：

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

1.23重複一次