LeetCode 66. Plus One

题目

题目大致意思是，给定一个数字字符数当做是一个数，然后给这个“数”加一，结果返回加一的数组。

分析

这道题类似小学做的那种加减法，分析一下几种情况：
1、该不该加一
2、加一进不进位的情况
3、进位后结果数组是否增加了
把这几个弄清楚其实代码就很好写了。

代码

class Solution {
    public int[] plusOne(int[] digits) {
       //申请数组的时候要考虑数组大小的问题，这里简单处理一下，只要末位要进位就多申请一个位置
       int num=digits[digits.length-1]+1==10?1:0;
       int[] result=new int[digits.length+num];
       //用flag记录要不要加1，
       int flag=1;
       for(int i=digits.length-1;i>=0;i--){
           if(flag==0){
           //不加1的时候，跟原来一样就行
            result[i]=digits[i];
          }
          if(flag==1){
          //要加1的时候判断一下要不要进位
              if(digits[i]+1==10){
                  if(i==0){
                  //如果已经进行到了第0个位置且第一个位置和为10的话，就要给第0个位置单独赋值了
                      result[0]=1;
                  }
                  result[i+1]=0;
              }
              else{
              //如果不用进位的话，只要在原来的基础上加1就行，同时标记为0，也就是不用再加1了
                  result[i]=digits[i]+1;
                  flag=0;
              }
          }

       }
       //之前我们多申请一个位置的时候，有可能用不上，这时候就需要将每一个数向前挪一个位置
        if(result.length!=digits.length&&result[0]!=1){
            int i=0;
            while(i<digits.length){
                result[i]=result[i++];
            }
            //同时将数组缩容
            result = Arrays.copyOf(result, result.length-1);
        }
        return result;
    }
}