LeetCode (2) - swift 实现数组加一

题目

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

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

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

示例 1:

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

示例 2:

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

class Solution {
    func plusOne(_ digits: [Int]) -> [Int] {
      if digits.isEmpty {
            return digits
        }
        
        //这里这样写的原因是，参数digits是let定义的不可变的哦
        var digits = digits
        
        for i in (0..<digits.count).reversed() {
            if digits[i] < 9 {
                digits[i] += 1
                return digits
            }
            digits[i] = 0
        }
        //运行至此说明carray还是等于1，即全是9的情况，需要在首位插入数字1
        digits.insert(1, at: 0)
        return digits
    }
}

func plusOne(_ digits:[Int]) -> [Int] {
    // 数组为空 直接返回
    if digits.isEmpty {
        return digits
    }
    
    var digits = digits
    for i in (0..<digits.count).reversed() {
        if digits[i] < 9 {
            digits[i] += 1
            return digits
        }
        digits[i] = 0
    }
    
    digits.insert(1, at: 1)
    print(digits)
    return digits
}

plusOne([1,2,3])

解题思路 :

1)  数组不存在直接retur
2) 对拿到的数组进行swift 反向遍历, 反向遍历是swift 的新特性
2.1) 如果0 位<9 直接将0 位加1 返回
2.2) 如果0 为为9 需要将0 为设置为0 , 如果1位为9 继续将1位设置+1 返回
2.3) 如果N位都为 0 ,就插入1  在第一位     digits.insert(1, at: 1)
 将元素插在数组的最左边

LeetCode (2) - swift 实现数组 加一

题目

示例 1:

示例 2:

LeetCode (2) - swift 实现数组加一