LeetCode之加一-Swift

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

示例

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

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

答案

每个元素的和若为10，则向前进一位，若数组中index=0的元素为0，则说明最高位进位1，在数组头部insert一个元素1（例如：数组[9,9,9],则进位成[1,0,0,0]）

代码如下：

func plusOne(_ digits: [Int]) -> [Int] {
        var result = digits
        var carry = 1

        for idx in result.indices {
            let newIdx = digits.count - 1 - idx
            let sum = digits[newIdx] + carry
            if sum == 10 {
                carry = 1
                result[newIdx] = 0
            } else {
                carry = 0
                result[newIdx] = sum
            }
        }
        if result.first == 0 {
            result.insert(1, at: 0)
        }
        return result
    }
复制代码