【Swift】LeetCode 189. 轮转数组
189. 轮转数组

题目描述

思路与 Swift 题解
解决这道题目的思路其实就是首先对整个数组进行翻转,然后翻转[0, k -1]和[k, n - 1]这个区间上的元素。注意,k要对数组的长度先原地求余。
因此,问题简化为,在 Swift 当中,如何对一个数组进行翻转。
Swift 当中并没有提供像 C++ 那种基于迭代器的区间式翻转,库函数提供了两种方法,分别是reverse()和reversed(),前者直接原地翻转,而后者返回翻转后的新数组。
看起来,Swift 的库函数并不能解决我们“按照区间”翻转的需求,但我们可以自己封装一个原地进行区间翻转的函数。我们定义一个新函数,它接受一个inout的数组,以及一个左闭右开的区间Range<Int>(完全闭区间的类型是ClosedRange<Int>,注意开区间和闭区间在 Swift 当中是不同的类型)。在这个函数当中,我们首先对区间指定的子数组进行一次翻转,之后再使用翻转后的区间替换原来数组当中相应的区间(可以使用库函数提供的replaceSubrange方法来完成原地替换)。
我们封装的函数定义为:
func reverseSubrange(_ nums: inout [Int], _ range: Range<Int>) {let reversedSubrange = Array(nums[range].reversed()) // 注意, 使用 reversed() 方法, 它将返回翻转后的数组.nums.replaceSubrange(range, with: reversedSubrange)// replaceSubrange 方法的签名就是 replaceSubrange(_:with:)
}
基于这个函数,我们可以实现我们刚才提到的思路并解决这道题,完整的 Swift 题解是:
class Solution {func reverseSubrange(_ nums: inout [Int], _ range: Range<Int>) {let reversedSubrange = Array(nums[range].reversed())nums.replaceSubrange(range, with: reversedSubrange)// replaceSubrange 方法的签名就是 replaceSubrange(_:with:)}func rotate(_ nums: inout [Int], _ k: Int) {var k = k % nums.countnums.reverse() // 对数组进行原地翻转reverseSubrange(&nums, 0..<k) // 翻转前 k 个元素reverseSubrange(&nums, k..<nums.count) // 翻转后 k 个元素}
}
