LeetCode-66-加一

题目要求

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

题目示例

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

题解

遇9变0，其余++，如果第一个元素为0，则前面加上一个1元素。

源代码

class Solution {
    public int[] plusOne(int[] digits) {
      int[] tmp = new int[digits.length];
	  boolean sign = false;
      int i = digits.length - 1;
      while(i >= 0){
        if(digits[i] + 1 > 9){
          tmp[i] = 0;
		  i--;
        }else{
          tmp[i] = digits[i] + 1;
		  i--;
		  sign = true;
          break;
        }
		
      }
      while(i >= 0){
        tmp[i] = digits[i];
		i--;
      }
	  if(sign) return tmp;
	  else{
		int[] tmp1 = new int[digits.length + 1];
		tmp1[0] = 1;
		return tmp1;
	  }
    }
}

改进后源代码

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