算法:旋转数组
2021-01-08 本文已影响0人
某非著名程序员
189. 旋转数组
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
示例 1:
输入: [1,2,3,4,5,6,7] 和 k = 3
输出: [5,6,7,1,2,3,4]
解释:
向右旋转 1 步: [7,1,2,3,4,5,6]
向右旋转 2 步: [6,7,1,2,3,4,5]
向右旋转 3 步: [5,6,7,1,2,3,4]
示例 2:
输入: [-1,-100,3,99] 和 k = 2
输出: [3,99,-1,-100]
解释:
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]
说明:
尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
要求使用空间复杂度为 O(1) 的 原地 算法。
思路一
- 先处理k=1的时候,向右移动一次。
- 循环k次,及移动k次。空间复杂度O(1),时间复杂度O(k%n*n)
func rotate(_ nums: inout [Int], _ k: Int) {
var k1 = k
if k == 0 {
return
}else if k>=nums.count{
k1 = k%nums.count
}
for _ in 0..<k1 {
var lastNum:Int = nums.last!
var curNum:Int!
for i in 0..<nums.count {
curNum = nums[i]
nums[i] = lastNum
lastNum = curNum
}
}
}
思路二
这道题让我想起了《剑指offer》中的面试题42:翻转单词顺序 VS 左旋转字符串。
基本思路就是先整体旋转,再局部旋转。先局部再整体也是可以的。
//92.68% 34.43%
func rotate(_ nums: inout [Int], _ k: Int) {
var k1 = k
if k == 0 {
return
}else if k>=nums.count{
k1 = k%nums.count
}
let rotateIndex = nums.count-k1-1
swapNums(&nums, 0, rotateIndex)
swapNums(&nums, rotateIndex+1, nums.count-1)
swapNums(&nums, 0, nums.count-1)
}
func swapNums(_ nums:inout [Int],_ i:Int,_ j:Int) {
if i>=j {
return
}
var i1 = i
var j1 = j
while i1<j1 {
let temp = nums[i1]
nums[i1] = nums[j1]
nums[j1] = temp
i1 += 1
j1 -= 1
}
}
总结
- 饮水思源,不求遇到新问题能做出来,只求遇到类似的问题能触类旁通。书不白看。