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
}
}