leetcode 初级之数组篇 07

加一

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

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

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

示例 1:

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

此种解法的执行效率战胜100%的Swift提交

class Solution {
    func plusOne(_ digits: [Int]) -> [Int] {
        var arr = digits
        var carry = 1
        var temp = 0
        let count = digits.count
        for (i, item) in digits.reversed().enumerated() {
            if carry == 0 {
                return arr
            }
            temp = item + carry
            carry = temp / 10
            arr[count - i - 1] = temp % 10
        }
        
        if carry != 0 {
            arr.insert(1, at: 0)
            return arr
        }
        return arr
    }
}