问题链接
问题描述
给你一个数组,将数组中的元素向右轮转 k
个位置,其中 k
是非负数。
示例
输入: nums = [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]
解题思路
- 解法1: 创建一个额外的数组,放置轮转之后的数。
- 解法2:运用数组反转的方法,例:
[1,2,3,4,5,6,7]. 向右轮转2个位置
A. 整个数组进行翻转,变为: [7,6,5,4,3,2,1]
B. 左边部分(第0位到第k-1位)进行翻转:[7,6]. =>. [6,7]
C.右边部分(第k位到最后一位)进行翻转:[5,4,3,2,1] => [1,2,3,4,5]
D.结果为:[6,7,1,2,3,4,5]
代码示例(JAVA)
创建额外的数组
class Solution {
public void rotate(int[] nums, int k) {
int[] result = new int[nums.length];
for (int i = 0; i < nums.length; i++) {
result[(i + k) % nums.length] = nums[i];
}
// for (int i = 0; i < nums.length; i++) {
// nums[i] = result[i];
// }
System.arraycopy(result, 0, nums, 0, nums.length);
}
}
数组翻转
class Solution {
public void rotate(int[] nums, int k) {
if (k % nums.length == 0) {
return;
}
reverse(nums, 0, nums.length - 1);
reverse(nums, 0, k % nums.length - 1);
reverse(nums, k % nums.length, nums.length - 1);
}
public void reverse(int[] nums, int start, int end) {
while (start < end) {
int temp = nums[start];
nums[start++] = nums[end];
nums[end--] = temp;
}
}
}