leetcode-加一

题目

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。

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

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

示例 1:

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

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

思路

我的思路很简单，没有什么技巧，主要分为三种情况讨论

1. 个位不等于9
2. 最高位和各位都等于9
3. 其他情况（我的第三个和第一个好像重复了，不管了，没什么差别）

我的解

/*
 * @lc app=leetcode.cn id=66 lang=java
 *
 * [66] 加一
 */

// @lc code=start
class Solution {
    public int[] plusOne(int[] digits) {



        if(digits[digits.length-1]!=9){
            digits[digits.length-1]+=1;
            return digits;
        }else if(digits[digits.length-1]==9 && digits[0] ==9){
            int a[] = new int[digits.length+1];

            int length = digits.length;
            int k;//进位
            k = (digits[length-1] +1)/10;
            digits[length-1] = ( digits[length-1] + 1 )%10;
            a[length] = digits[length-1];
            for(int i = length-2;i>=0;i--){

                k = (digits[i]+k)/10;

                digits[i] = (digits[i]+k)%10;
                a[i+1] = digits[i];

                
            }
            a[0] = k;
            return a;
        }else{
            int length = digits.length;
            int k;//进位
            k = (digits[length-1] +1)/10;
            digits[length-1] = ( digits[length-1] + 1 )%10;
            
            for(int i = length-2;i>=0;i--){
                int whw = digits[i] +k;
                digits[i] =whw%10;
                k = whw/10;
                
            }
            
            return digits;
        }





        


    }
}
// @lc code=end

我的结果

Accepted
109/109 cases passed (0 ms)
Your runtime beats 100 % of java submissions
Your memory usage beats 5.14 % of java submissions (38.5 MB)

其他解答

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