加一

问题描述

加一

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

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

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

示例 1:

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

解决方案

模拟加法过程。

class Solution {
    public int[] plusOne(int[] digits) {
        int pos=digits.length-1;//正在计算的位置
        digits[pos]++;
        while (digits[pos]>9) {
            digits[pos]-=10;
            pos--;
            if(pos<0)break;
            digits[pos]++;  
        }
        if(pos>=0) {
            return digits;
        }
        else {
            int[] newa=new int[digits.length+1];
            for (int i =0; i < digits.length; i++) {
                newa[i+1] = digits[i];              
            }
            newa[0]=1;
            return newa;
        }
    }
}

或者如下Java代码更为通用（大树相加）：

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