LintCode之加一

题目描述：

 

分析：由样例可以知道，当数组的每一个数字都是9时，加一会产生一个最高位的数字1，所以先判断这个数组的每一位是否都是9，如果是，那么新数组的大小是原数组大小加一，否则新数组的大小等于原数组的大小。数字加一，其实也就是数组的最后一个数字加一，但是加一的时候可能会产生进位（9+1=10：就产生了向前一位的进位），用一个数字k表示向前一位的进位（k=1:向前产生了进位，k=0:没有产生进位）由此加到原数组的第一个数字为止，如果最高位也就是原数组的第一位也产生了进位，那么将新数组的第一位置为1。

代码：

public class Solution {
    /*
     * @param digits: a number represented as an array of digits
     * @return: the result
     */
    public int[] plusOne(int[] digits) {
        // write your code here
        if(digits.length == 0 || digits == null) {
            return null;
        }
        int[] num = null;
        
        int i;
        //判断数组的每位数字是否都是9
        for(i=0; i<digits.length; i++) {
            if(digits[i] != 9) {
                break;
            }
        }
        if(i == digits.length) {
            num = new int[digits.length+1];
        }else {
           num = new int[digits.length];
        }

        int n = num.length-1;
        //k表示低位向高位的进位
        int k = 1;
        for(int j=digits.length-1; j>=0; j--) {
            num[n--] = (digits[j]+k)%10;
            k = (digits[j]+k)/10;
        }
        if(k == 1) {
            num[n] = k;
        }
        
       
        
        return num;
    }
}