题目描述
将包含 n 个元素的数组向右旋转 k 步。
例如,如果 n = 7
, k = 3
,给定数组 [1,2,3,4,5,6,7]
,向右旋转后的结果为 [5,6,7,1,2,3,4]
。
注意:
尽可能找到更多的解决方案,这里最少有三种不同的方法解决这个问题。
提示:
要求空间复杂度为 O(1)
https://github.com/biezhihua/LeetCode
解法1
借助
O(n)
O
(
n
)
的空间解法。再利用(i + k) % n = 旋转后的位置
,可以将原有数组中的数据复制到新数组中。
例如:
3元素在数组中的2位置上,(2+3) % 7 = 5
5元素在数组中的4位置上,(4+3) % 7 = 0
6元素在数组中的5位置上,(5+3) % 7 = 1
7元素在数组中的6位置上,(6+3) % 7 = 2
根据以上可以推断出,诸如将一个数组向右偏移或者旋转多少步等之类的题目,可以通过(元素位置+偏移或旋转步数) 模与 数组长度
可以得到偏移或者旋转之后的元素位置
。
代码:
@Test
public void test() {
Assert.assertArrayEquals(new int[]{1, 2, 3, 4, 5, 6, 7}, rotate(new int[]{1, 2, 3, 4, 5, 6, 7}, 0));
Assert.assertArrayEquals(new int[]{7, 1, 2, 3, 4, 5, 6}, rotate(new int[]{1, 2, 3, 4, 5, 6, 7}, 1));
Assert.assertArrayEquals(new int[]{6, 7, 1, 2, 3, 4, 5}, rotate(new int[]{1, 2, 3, 4, 5, 6, 7}, 2));
Assert.assertArrayEquals(new int[]{5, 6, 7, 1, 2, 3, 4}, rotate(new int[]{1, 2, 3, 4, 5, 6, 7}, 3));
Assert.assertArrayEquals(new int[]{4, 5, 6, 7, 1, 2, 3}, rotate(new int[]{1, 2, 3, 4, 5, 6, 7}, 4));
Assert.assertArrayEquals(new int[]{3, 4, 5, 6, 7, 1, 2}, rotate(new int[]{1, 2, 3, 4, 5, 6, 7}, 5));
Assert.assertArrayEquals(new int[]{2, 3, 4, 5, 6, 7, 1}, rotate(new int[]{1, 2, 3, 4, 5, 6, 7}, 6));
Assert.assertArrayEquals(new int[]{1, 2, 3, 4, 5, 6, 7}, rotate(new int[]{1, 2, 3, 4, 5, 6, 7}, 7));
Assert.assertArrayEquals(new int[]{2, 1}, rotate(new int[]{1, 2}, 1));
Assert.assertArrayEquals(new int[]{2, 1}, rotate(new int[]{1, 2}, 3));
}
public int[] rotate(int[] nums, int k) {
k = k % nums.length;
int[] result = new int[nums.length];
for (int i = 0; i < nums.length; i++) {
result[(i + k) % nums.length] = nums[i];
}
return result;
}
解法2
可以采取反转的方法,先反转前n-k
个元素,再反转后k
个元素,然后再将整个数组反转,就能得到该数组旋转k
个元素的结果了
@Test
public void test1() {
Assert.assertArrayEquals(new int[]{7, 1, 2, 3, 4, 5, 6}, rotate1(new int[]{1, 2, 3, 4, 5, 6, 7}, 1));
Assert.assertArrayEquals(new int[]{1, 2, 3, 4, 5, 6, 7}, rotate1(new int[]{1, 2, 3, 4, 5, 6, 7}, 0));
Assert.assertArrayEquals(new int[]{6, 7, 1, 2, 3, 4, 5}, rotate1(new int[]{1, 2, 3, 4, 5, 6, 7}, 2));
Assert.assertArrayEquals(new int[]{5, 6, 7, 1, 2, 3, 4}, rotate1(new int[]{1, 2, 3, 4, 5, 6, 7}, 3));
Assert.assertArrayEquals(new int[]{4, 5, 6, 7, 1, 2, 3}, rotate1(new int[]{1, 2, 3, 4, 5, 6, 7}, 4));
Assert.assertArrayEquals(new int[]{3, 4, 5, 6, 7, 1, 2}, rotate1(new int[]{1, 2, 3, 4, 5, 6, 7}, 5));
Assert.assertArrayEquals(new int[]{2, 3, 4, 5, 6, 7, 1}, rotate1(new int[]{1, 2, 3, 4, 5, 6, 7}, 6));
Assert.assertArrayEquals(new int[]{1, 2, 3, 4, 5, 6, 7}, rotate1(new int[]{1, 2, 3, 4, 5, 6, 7}, 7));
Assert.assertArrayEquals(new int[]{2, 1}, rotate1(new int[]{1, 2}, 1));
Assert.assertArrayEquals(new int[]{2, 1}, rotate1(new int[]{1, 2}, 3));
}
public int[] rotate1(int[] nums, int k) {
reverse(nums, 0, nums.length - 1 - k);
reverse(nums, nums.length - k, nums.length - 1);
reverse(nums, 0, nums.length - 1);
return nums;
}
private void reverse(int[] nums, int start, int end) {
while (start < end) {
int tmp = nums[start];
nums[start++] = nums[end];
nums[end--] = tmp;
}
}
解法3
依旧可以利用(i+k)%n
等于新i
的思路,不过这次是每次调换一个元素,后一个元素的调换基于上一个的位置。
例如:让数组[1,2,3,4,5,6,7]
向右旋转1,可以按照以下步骤来:
// 1
// 1, 2, 3, 4, 5, 6, 7
// 3
// 1, 2, 1, 4, 5, 6, 7
// 5
// 1, 2, 1, 4, 3, 6, 7
// 7
// 1, 2, 1, 4, 3, 6, 5
// 2
// 1, 7, 1, 4, 3, 6, 5
// 4
// 1, 7, 1, 2, 3, 6, 5
// 6
// 1, 7, 1, 2, 3, 4, 5
// 1
// 6, 7, 1, 2, 3, 4, 5
public void rotate2(int[] nums, int k) {
if (nums.length == 0 || (k %= nums.length) == 0) {
return;
}
int length = nums.length;
int start = 0;
int i = 0;
int cur = nums[i];
int cnt = 0;
while (cnt++ < length) {
i = (i + k) % length;
int t = nums[i];
nums[i] = cur;
if (i == start) {
++start;
++i;
cur = nums[i];
} else {
cur = t;
}
}
}