题目:
给你一个数组,将数组中的元素向右轮转 k 个位置,其中 k 是非负数。
示例 1:
输入: 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]
示例 2:
输入:nums = [-1,-100,3,99], k = 2
输出:[3,99,-1,-100]
解释:
向右轮转 1 步: [99,-1,-100,3]
向右轮转 2 步: [3,99,-1,-100]
提示:
1 <= nums.length <= 10^5
-2^31 <= nums[i] <= 2^31 - 1
0 <= k <= 10^5
进阶:
尽可能想出更多的解决方案,至少有 三种 不同的方法可以解决这个问题。
你可以使用空间复杂度为 O(1) 的 原地 算法解决这个问题吗?
解法1:新建数组
/**
* 思路:
* 新建数组,把当前数组元素全部放入新数组
* 遍历新数组,给每一位都移动k位,赋值到原数组
*/
public static void rotate(int[] nums, int k) {
int[] new_arr = new int[nums.length];
System.arraycopy(nums,0,new_arr,0,nums.length);
k%=nums.length;
for (int i=0;i<nums.length;i++){
if (i+k<nums.length) {
nums[i + k] = new_arr[i];
}else {
nums[(i + k)%nums.length] = new_arr[i];
}
}
}
时间复杂度:On
空间复杂度:On
解法2:移动k次
首位交换
/**
* 思路:
* 一次移动一位,移动k次
* 不断的让当前值和第一个值进行换位。最后的结果就是移动1的结果
*/
public static void rotate(int[] nums, int k) {
k%=nums.length;
for (int i=0;i<k;i++){
for (int j=0;j<nums.length-1;j++){
int tmp = nums[j + 1];
nums[j+1]=nums[0];
nums[0]=tmp;
}
}
}
时间复杂度:On^2
空间复杂度:O1
末尾交换
/**
* 思路:
* 一次移动一位,移动k次
* 不断的让当前值和最后一个值进行换位。最后的结果就是移动1的结果
*/
public static void rotate(int[] nums, int k) {
for (int i=0;i<k;i++){
for (int j=0;j<nums.length-1;j++){
int tmp=nums[j];
nums[j]=nums[nums.length-1];
nums[nums.length-1]=tmp;
}
}
}
时间复杂度:On^2
空间复杂度:O1
新建数组
/**
*思路:
* 深拷贝一个数组,把数组的当前位置值赋值给下个位置。如果是数组的末尾就赋值给第一个位置
* 重复k次
*/
public static void rotate(int[] nums, int k) {
for (int i=0;i<k;i++){
int[] copy_nums = nums.clone();
for (int j=0;j<nums.length;j++){
if (j==nums.length-1){
nums[0]=copy_nums[j];
} else {
nums[j+1]=copy_nums[j];
}
}
}
}
时间复杂度:On^2
空间复杂度:On
解法3:3次反转
/**
* 思路:
* 3次反转数组:
* 先整体反转
* 反转前k个(整体反转后的前k个元素,就是原数组中会替换到数组头的元素)
* 反转后n-k个(反转后n-k个,因为前面有k个元素,反转后相当于移动了k位)
*
*
* 原始数组 : 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 --> 结果
*/
public void rotate(int[] nums, int k) {
k%=nums.length;
reverse(nums,0,nums.length-1);
reverse(nums,0,k-1);
reverse(nums,k,nums.length-1);
}
private void reverse(int[] nums, int start, int end) {
while (end>start){
int tmp = nums[start];
nums[start++]=nums[end];
nums[end--]=tmp;
}
}
时间复杂度:On
空间复杂度:O1