题目描述:
给定一个数组,将数组中的元素向右移动 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]
package com.zhou.rotate;
import java.util.Arrays;
/**
* FileName: Rotate
*
* @author Mozzie
* @date 2020/11/29 17:09
* @Description 旋转数组
*/
public class Rotate {
public static void main(String[] args) {
int[] nums = {1, 2, 3, 4, 5, 6};
rotateSolutionTwo(nums, 2);
System.out.println(Arrays.toString(nums));
}
/**
* 暴力解法
* 共循环 k 次,复制数组从先记录最后一位,然后依次覆盖复制,再讲最后一位放置第一位上
* 若k = 3,nums = [1,2,3,4,5,6,7]
* 第一次操作:记录最后一位为7,进行复制操作完后数组为[1,1,2,3,4,5,6],替换第一位后数组为:[7,1,2,3,4,5,6]
* 第二次操作:记录最后一位为6,进行复制操作完后数组为[7,7,1,2,3,4,5],替换第一位后数组为:[6,7,1,2,3,4,5]
* 第三次操作:记录最后一位为5,进行复制操作完后数组为[6,6,7,1,2,3,4],替换第一位后数组为:[5,6,7,1,2,3,4]
*
* @param nums 待旋转数组
* @param k 右移位置
*/
public static void rotateSolutionOne(int[] nums, int k) {
for (int i = 0; i < k; i++) {
int lastNum = nums[nums.length - 1];
for (int j = nums.length - 1; j > 0; j--) {
nums[j] = nums[j - 1];
}
nums[0] = lastNum;
}
}
/**
* 使用反转
* 若k = 3,nums = [1,2,3,4,5,6,7]
* 第一次反转数组:nums = [7,6,5,4,3,2,1]
* 第二次反转数组,反转范围为前k-1个,即为前3个:nums = [5,6,7,4,3,2,1]
* 第三次反转数组,反转范围为k至nums.length - 1,即为后4个:nums = [5,6,7,1,2,3,4]
*
* @param nums 待旋转数组
* @param k 右移位置
*/
public static void rotateSolutionTwo(int[] nums, int k) {
k %= nums.length;
reverse(nums, 0, nums.length - 1);
reverse(nums, 0, k - 1);
reverse(nums, k, nums.length - 1);
}
/**
* 反转数组(首尾替换)
*
* @param nums 待反转数组
* @param start 开始
* @param end 结束
*/
public static void reverse(int[] nums, int start, int end) {
while (start < end) {
int temp = nums[start];
nums[start] = nums[end];
nums[end] = temp;
start++;
end--;
}
}
}