leetcode： 加一（scala）

题目

给定一个由整数组成的非空数组所表示的非负整数，在该数的基础上加一。

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

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

思路

遇到9要进一位，每一个位置只能填一个数字

scala

object Solution {
  def plusOne(digits: Array[Int]): Array[Int] = {
    var digit = digits 
    val len = digit.length
    var i = len - 1
    digit(i) += 1
    if (digit(i) > 9) {
      while (i > 0 && digit(i) > 9) {
        digit(i) = 0
        i = i - 1
        digit(i) += 1
      }

      if (i == 0 && digit(i) > 9) {
        digit = new Array[Int](len + 1)
        digit(0) = 1
      }
    }
    digit
  }
}

java

class Solution {
    public int[] plusOne(int[] digits) {
        int len = digits.length;
        int i = len - 1;
        digits[i] += 1;
        if (digits[i] > 9) {
            while (i > 0 && digits[i] > 9) {
                digits[i] = 0;
                digits[--i] += 1;
            }

            if (i == 0 && digits[i] > 9) {
                digits = new int[len + 1];
                digits[0] = 1;
            }
        }
        return digits;
    }
}