[LeetCode] 每日一题 Day 1：66. 加一

题目

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。
最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

输入: [1,2,3]
输出: [1,2,4]
解释: 输入数组表示数字 123。

示例 2:

输入: [4,3,2,1]
输出: [4,3,2,2]
解释: 输入数组表示数字 4321。

解决

思路：

设置进位为1，从末尾开始逐渐累加，一旦进位为0，直接return，如果遍历完了进位还为1，说明要扩充一个位置，这时新建一个数组重新赋值进去

代码

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

复杂度分析

时间复杂度：最好情况O(1)，最坏情况O(n)
空间复杂度：O(n)