leetcode66. 加一

題目描述：

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

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

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

示例 1:

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

示例 2:

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

思路：

包含加法運算，一定會有溢出情況的測試用例。所以一開始就杜絕使用加法進行解決的思路。最基礎的模擬加法操作，滿9進1。然後就是特殊的全是9的情況，這樣必須要重新申請數組，所以單獨拿出來考慮。

class Solution {
	public int[] plusOne(int[] digits) {
        // 9999...特殊情況解決
        int i, count = 0;
        int[] ans = new int[digits.length + 1];
        for(i = 0; i < digits.length; i++)
            if(digits[i] == 9) count++;
        if(count == digits.length){
            ans[0] = 1;
            return ans;
        }
        return ans(digits, digits.length - 1); // 遞歸模擬滿9進1
	}
    
    public int[] ans(int[] digits, int index){           
        if(digits[index] < 9){
            digits[index]++;
            return digits;
        }
        digits[index] = 0;  
        return ans(digits, index - 1);
    }
}

不過遞歸佔用的內存是比較多的，將遞歸改爲循環會更好一點：

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