LeetCode-66-加一

题目跳转

题目描述

给定一个由 整数 组成的 非空 数组所表示的非负整数，在该数的基础上加一。
最高位数字存放在数组的首位， 数组中每个元素只存储单个数字。
你可以假设除了整数 0 之外，这个整数不会以零开头。

• 示例1
  输入：digits = [1,2,3]
  输出：[1,2,4]
  解释：输入数组表示数字 123。
• 示例2
  输入：digits = [4,3,2,1]
  输出：[4,3,2,2]
  解释：输入数组表示数字 4321。
• 示例3
  输入：digits = [0]
  输出：[1]
• 提示
  1 <= digits.length <= 100
  0 <= digits[i] <= 9

思路

可以分三种情况讨论

1. digits最后面的数字不是9，则可以将其加1进行返回
2. digits最后面的数字为9，则逆向向前查找，找到第一个不为9的数字，将其加1，然后将其后面为9的数字都置0，进行返回
3. digits全部数字都为9，则需要构造一个新数组，长度比digits多1，将其首位置1，进行返回。

代码

class Solution {
    public int[] plusOne(int[] digits) {
       int n=digits.length;
       for(int i=n-1;i>=0;i--){
           if(digits[i]!=9){
               digits[i]++;
               for(int j=i+1;j<n;j++){
                   digits[j]=0;
               }
               return digits;
           }
       }
       //如果digits中所有的元素都是9，需要返回一个长度为n+1的新数组
       int []  ans=new int[n+1];
       ans[0]=1;
       return ans;
    }
}

第一次写博客，大佬勿喷，记录一下题解