题目: 旋转数组
给定一个数组,将数组中的元素向右移动 k 个位置,其中 k 是非负数。
复制代码
示例:
输入: [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,-100,3,99] 和 k = 2
输出: [3,99,-1,-100]
解释:
向右旋转 1 步: [99,-1,-100,3]
向右旋转 2 步: [3,99,-1,-100]
复制代码
思考:
这道题有一种巧妙地利用反转的作法。
首先将第0个到第k个元素反转,再将第k+1到末尾元素反转,最后再将所有元素反转便可。
例如:[1,2,3,4,5,6,7] k = 3
将0到3反转:[4,3,2,1,5,6,7]
将4到6反转:[4,3,2,1,7,6,5]
所有翻转:[5,6,7,1,2,3,4] 获得最后结果。
这里要注意下还有这样的状况:[1,2] k = 5 即k大于数组长度的状况。
这里能够发现数组旋转次数等于数组长度时,旋转后的数组与初始数组相同,转了一圈又回来了。
1次旋转:[2,1]
2次旋转: [1,2] 转回来了
3次旋转:[2,1]
4次旋转: [1,2] 转回来了
5次旋转:[2,1]
因此这里的有效k等于k对数组长度求余。
复制代码
实现:
class Solution {
public void rotate(int[] nums, int k) {
int length = nums.length;
k %= length;
reverse(nums, 0, length - 1);
reverse(nums, 0, k - 1);
reverse(nums, k, length - 1);
}
private void reverse(int[] nums, int start, int end) {
while (start < end) {
int temp = nums[start];
nums[start++] = nums[end];
nums[end--] = temp;
}
}
}复制代码