Leetcode189. 旋转数组
题目:
给定一个数组,将数组中的元素向右移动 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]
题解:
方案一:使用额外的数组
单独定义一个和原始数组长度相同的数组,旋转后的数据写到新的数组,当旋转结束再把值拷贝回原始数组。
- 时间复杂度: O ( n ) O(n) O(n) 。将数字放到新的数组中需要一遍遍历,另一边来把新数组的元素拷贝回原数组。
- 空间复杂度: O ( n ) O(n) O(n)。另一个数组需要原数组长度的空间。
方案二:使用环状替换
- 时间复杂度: O ( n ) O(n) O(n) 。只遍历了每个元素一次。
- 空间复杂度:
O
(
1
)
O(1)
O(1) 。使用了常数个额外空间。
方案三:反转
例:
原始数组 : 1 2 3 4 5 6 7
反转所有数字后 : 7 6 5 4 3 2 1
反转前 k 个数字后 : 5 6 7 4 3 2 1
反转后 n-k 个数字后 : 5 6 7 1 2 3 4 --> 结果 - 时间复杂度: O ( n ) O(n) O(n)。 n n n 个元素被反转了总共 3 次。
- 空间复杂度: O ( 1 ) O(1) O(1) 。 没有使用额外的空间。
scala代码如下:
/**
* 使用额外的数组
* @param nums
* @param k
*/
def rotate(nums: Array[Int], k: Int): Unit = {
val arr = new Array[Int](nums.length)
for (i <- 0 until nums.length) {
arr((i + k) % nums.length) = nums(i)
}
for (i <- 0 until nums.length) {
nums(i) = arr(i)
}
}
/**
* 使用环状替换
*
* @param nums
* @param k
*/
def rotate2(nums: Array[Int], k: Int): Unit = {
var k1 = k
k1 = k1 % nums.length
var count = 0
var start = 0
while (count < nums.length) {
var current = start
var pre = nums(start)
do {
val next = (current + k1) % nums.length
val temp = nums(next)
nums(next) = pre
pre = temp
current = next
count = count + 1
} while (start != current)
start = start + 1
}
}
/**
* 反转
* 原始数组 : 1 2 3 4 5 6 7
* 反转所有数字后 : 7 6 5 4 3 2 1
* 反转前 k 个数字后 : 5 6 7 4 3 2 1
* 反转后 n-k 个数字后 : 5 6 7 1 2 3 4 --> 结果
*
* @param nums
* @param k
*/
def rotate3(nums: Array[Int], k: Int): Unit = {
var k1 = k
k1 = k1 % nums.length
reverse(nums, 0, nums.length - 1)
reverse(nums, 0, k1 - 1)
reverse(nums, k1, nums.length - 1)
}
def reverse(nums: Array[Int], start: Int, end: Int): Unit = {
var start1 = start
var end1 = end
while (start1 < end1) {
val temp = nums(start1)
nums(start1) = nums(end1)
nums(end1) = temp
start1 = start1 + 1
end1 = end1 - 1
}
}