leetcode刷题笔记六十六题 加一

leetcode刷题笔记六十六题 加一

源地址：66. 加一

问题描述：

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

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

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

示例 1:

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

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

代码补充：

/**
本题较为简单，可分三种情况考虑
1.针对末位数字不为9时，直接对数组最后一位加数字1
2.针对末尾数字为9，但是不全为9时，不为9的位置加1并弹出，为9的位置置0
3.针对全部为9的状态，这种情况下，数组中存放的结果必为首位为0，依题目可知，这种做法并不合理，即说明存在进位1需要补充
*/
import scala.collection.mutable
import util.control.Breaks._
object Solution {
    def plusOne(digits: Array[Int]): Array[Int] = {
        val length = digits.length
        if (digits(length-1) != 9) digits(length-1) += 1
        else{
            breakable{
                for ( i <- (0 to length-1).reverse){
                    if(digits(i) != 9){
                        digits(i) += 1
                        break()
                    }
                    digits(i) = 0
                }
                if(digits(0) == 0){
                    return(Array[Int](1).concat(digits))
                }
            }

        }
        
        return digits
    }
}