題目

題目連結：https://leetcode.cn/problems/plus-one/

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

最高位數字存放在陣列的首位， 陣列中每個元素只儲存單個數字。

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

示例 1：

輸入：digits = [1,2,3]
輸出：[1,2,4]
解釋：輸入陣列表示數字 123。

示例 2：

輸入：digits = [4,3,2,1]
輸出：[4,3,2,2]
解釋：輸入陣列表示數字 4321。

示例 3：

輸入：digits = [0]
輸出：[1]

題解

官方

思路

當我們對陣列 \(\textit{digits}\)

如果 \(\textit{digits}\) 的末尾沒有 999，例如 [1,2,3]，那麼我們直接將末尾的數加一，得到 [1,2,4] 並返回；

如果 \(\textit{digits}\) 的末尾有若干個 999，例如 [1,2,3,9,9]，那麼我們只需要找出從末尾開始的第一個不為 999 的元素，即 333，將該元素加一，得到 [1,2,4,9,9]。隨後將末尾的 999
全部置零，得到 [1,2,4,0,0] 並返回。

如果 \(\textit{digits}\)

演算法

我們只需要對陣列 \(\textit{digits}\) 進行一次逆序遍歷，找出第一個不為 999 的元素，將其加一併將後續所有元素置零即可。如果 \(\textit{digits}\) 中所有的元素均為999，那麼對應著「思路」部分的第三種情況，我們需要返回一個新的陣列。

class Solution {
    public int[] plusOne(int[] digits) {
        int n = digits.length;
        for (int i = n - 1; i >= 0; --i) {
            if (digits[i] != 9) {
                ++digits[i];
                for (int j = i + 1; j < n; ++j) {
                    digits[j] = 0;
                }
                return digits;
            }
        }

        // digits 中所有的元素均為 9
        int[] ans = new int[n + 1];
        ans[0] = 1;
        return ans;
    }
}
 

方法二

官網寫的稍微複雜些，不判斷當前值，判斷加1後的末尾是不是0，程式碼如下:

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