[LeetCode] Plus One

题目链接：https://leetcode.com/problems/plus-one/

 

Given a non-negative number represented as an array of digits, plus one to the number.

The digits are stored such that the most significant digit is at the head of the list.

 

针对本题可能存在的疑惑

1）数字的头部能否包含0（例如 [0,0,1]）？

 

解题思路：

从低位向高位递归，存在如下两种情况

1）若第i位的数 + 1 <= 9，完成加法后直接返回

2）若第i位的数 + 1 = 10（即若第i位的数为9），完成加法后第i位为0，循环到第i-1位

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

时间复杂度：O(n) ，空间复杂度：O(n)

 

扩展：可以用List替代数组操作

     public void plusOneList(List<Integer> digits) {
        for(int i = digits.size() - 1; i >= 0; i--){
            int digit = digits.get(i);
            if(digit < 9){
                digits.set(i, digit + 1);
                return;
            }else{
                digits.set(i, 0);
            }    
        }
        digits.add(0);
        digits.set(0, 1);        
    }

时间复杂度：O(n) ，空间复杂度：O(1)