Leetcode - 旋转数组

问题描述

给定一个数组,将数组中的元素向右移动 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]

说明

尽可能想出更多的解决方案,至少有三种不同的方法可以解决这个问题。
要求使用空间复杂度为 O(1) 的原地算法。

解答

one

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
@Test
public void testRotate() {
int[] arrays = {1};
rotate(arrays, 3);
for (int i = 0; i < arrays.length; i++) {
System.out.print(arrays[i]);
}
}

public static void rotate(int[] arrays, int k) {
if (0 >= k) {
return;
}
for (int i = 0; i < k; i++) {
int last = arrays[arrays.length - 1];
int first = arrays[0];
for (int j = 1; j < arrays.length; j++) {
last = arrays[j];
arrays[j] = first;
first = last;
}
arrays[0] = last;
}


}

two

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
public void rotate(int[] nums, int k) {
if (nums.length == 0) {
return;
}

k = k % nums.length;
reverse(nums, 0, nums.length - k - 1);
reverse(nums, nums.length - k, nums.length - 1);
reverse(nums, 0, nums.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;
start++;
end--;
}
}

three

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
/**
* (k+i)%n 计算出新的位置
*/
public void rotate(int[] nums, int k) {
k = k % nums.length;
if(k ==0 || nums.length == 1)
return;
int N=nums.length;
int n=0;
int i=0;
int times=0;
int cur=nums[0];
// n==N out
while(n<N){
// 计算的i 可能和之前的冲突,移到下一个i,再计算
i=(k+i)%N;
int t = nums[i];
nums[i] = cur;
if(i==times){
times++;
i++;
cur = nums[i];
}else{
cur = t;
}
n++;
}
}

Powered by Hexo

Copyright © 2016 - 2019 When I think of you, I smile. All Rights Reserved.

UV : | PV :