【LintCode题解】Q407

[Q407]

描述：

　　给定一个非负数，表示一个数字数组，在该数的基础上+1，返回一个新的数组。

　　该数字按照大小进行排列，最大的数在列表的最前面。

　　样例：

　　给定 [1,2,3] 表示 123, 返回 [1,2,4].

　　给定 [9,9,9] 表示 999, 返回 [1,0,0,0].

思路:

　　此题目第一眼感觉是将数组中的元素取出转化为一个int型数，然后将其加一，再拆解这个数存储回数组中，想了想发现这种算法太麻烦。

　　其实这个题目只存在两种情况：

　　1 如果这个数是9，99，999，9999之类的全是9的，需要将数组长度增加一，第一个元素是1，剩余元素为0.

　　2 如果这个数不需要增加数组长度，只需要找到第一个不是9的元素将其加一即可，在遍历的过程中，遇到的为9的元素，直接将其置为0

实现：

public int[] plusOne(int[] digits) {
　　　　　
        int flag = 0 ;
        for(int i = 0 ; i < digits.length ; i ++){
            if(digits[i] == 9){
                flag += 1;
            }
            else{
                continue;
            }
        }
        if(flag == digits.length){
            int[] update = new int[digits.length + 1];
            update[0] = 1;
            for(int i = 1 ; i < update.length ; i ++){
                update[i] = 0 ;
            }
            return update;
        }
        else{
            for(int i = digits.length - 1 ; i >= 0 ; i --){

                if(digits[i] != 9){
                    digits[i] += 1 ;
                    break;
                }
                else {
                    digits[i] = 0 ;
                }
            }
            return digits;
        }

}