Leetcode-plus-one

题目描述

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

一个数看做一个数组，对这个数加1，求结果数组。

我的方法貌似有点蠢。

public class Solution {
    public int[] plusOne(int[] digits) {
        int jingwei = 0;
        int n = digits.length;

        int[] res = new int[n+1];
        
        if(digits[n-1]+1 == 10) {
            jingwei = 1;
            res[n] = 0;
        }else{
            res[n] += digits[n-1] + 1;
        }

        if(n == 1 && jingwei == 1) {
            res[0] = 1;
        }
        for(int i=n-2; i>=0; i--) {
            res[i+1] = ( digits[i]+jingwei ) % 10;
            jingwei = ( digits[i]+jingwei ) / 10;
        	if(i == 0 && jingwei == 1) {
                res[0] = 1;
            }
        }
        if(res[0] == 0) {
            int[] res1 = new int[n];
            for(int i=0; i<n; i++) {
                res1[i] = res[i+1];
            }
            return res1;
        }else{
            return res;
        }
    }
}

看了一个别人的方法：

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

一般情况先return，不满足一般情况的唯一特殊情况就是最终结果是10,100,1000这种，那么我们最后new一个新数组，并把第一个元素置为1，然后return即可。