Rotate Array
LeetCode https://leetcode.cn/problems/rotate-array/
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
29
30
class Solution {
public void rotate(int[] nums, int k) {
// index 0 1 2 3 4 5 6
// nums 1 2 3 4 5 6 7
// step1 7 6 5 4 3 2 1
// step2 5 6 7 4 3 2 1
// step3 5 6 7 1 2 3 4
// result 5 6 7 1 2 3 4
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 left, int right) {
if (left >= right) {
return;
}
exchange(nums, left, right);
reverse(nums, left + 1, right - 1);
}
// exchange arr[i] and a[j]
private void exchange(int[] arr, int i, int j) {
int swap = arr[i];
arr[i] = arr[j];
arr[j] = swap;
}
}
iterative reversal
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
class Solution {
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 left, int right) {
while (left < right) {
exchange(nums, left++, right--);
}
}
// exchange arr[i] and a[j]
private void exchange(int[] arr, int i, int j) {
int swap = arr[i];
arr[i] = arr[j];
arr[j] = swap;
}
}
1
2
3
4
5
6
7
private void reverse(int[] nums, int left, int right) {
while (left < right) {
exchange(nums, left, right);
left++;
right--;
}
}
Complexity
- Time = O(n)
- Space = O(1)
This post is licensed under CC BY 4.0 by the author.