LeetCode - 加一

一. 题目描述

给定一个非负整数组成的非空数组，在该数的基础上加一，返回一个新的数组。

最高位数字存放在数组的首位， 数组中每个元素只存储一个数字。

你可以假设除了整数 0 之外，这个整数不会以零开头。

示例 1:

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

示例 2:

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

二. 我的解法

2.1 代码

class Solution {
    public int[] plusOne(int[] digits) {
        int[] result;
        int i;
        for( i = digits.length - 1; i >= 0; i-- )
        {
            if( digits[i] == 9 )
            {
                digits[i] = 0;
                continue;
            }
            else
            {
                digits[i] += 1;
                break;
            }

        }
        if( i < 0 && digits[0] == 0 )
        {
            result = new int[digits.length+1];
            result[0] = 1;
            for( i = 0; i < digits.length; i++ )
            {
                result[i+1] = digits[i];
            }
        }
        else
        {
            result = digits;
        }
        return result;
    }
}

2.2 分析

1. 一开始以为就是末位加一就ok了， 然后发现才发现是需要进位的，“一个位置只存储一个数字”是这个意思。

2. 第一个for循环是为了加一

3. 接下来的 if 语句是为了判断是否需要增加数字的位数，即从原来的4为进到5位，那么如果需要进位的话，新加的一位必然为1，原来的首位必然为0。

三. 更优解法

3.1 代码

class Solution {
    public int[] plusOne(int[] digits) {
        boolean add = true;
        int len = digits.length;
        for (int i = len - 1; i > -1; i--) {
            if (digits[i] == 9 ) {
                digits[i] = 0;
            } else {
                digits[i] += 1;
                return digits;
            } 
        }
        
        int[] newArray = new int[len + 1];
        newArray[0] = 1;
        return newArray;
    }
}

3.2 分析

1. 前面的思路都是一样的，不同的地方在于增加位数的地方

2. 很明显只有当原数字为9，99， 999， 9999.....时， 才会有增加位数的可能，而一旦增加了位数，新的数字必为10， 100， 1000， 10000.....，那么，增加位数的代码就很清晰了。----这么明显的设定，我还是没有发现呀，脑瓜跟不上呀。

四. 参考

https://leetcode-cn.com/problems/plus-one/description/