71 lines
1.6 KiB
Java
71 lines
1.6 KiB
Java
import java.util.*;
|
|
|
|
public class ArrayRotate {
|
|
|
|
public static void rotate1(int[] a, int first, int middle, int last) {
|
|
int next = middle;
|
|
while (first != next) {
|
|
swap(a, first++, next++);
|
|
if (next == last)
|
|
next = middle;
|
|
else if (first == middle)
|
|
middle = next;
|
|
}
|
|
}
|
|
|
|
public static void rotate2(int[] a, int first, int middle, int last) {
|
|
reverse(a, first, middle);
|
|
reverse(a, middle, last);
|
|
reverse(a, first, last);
|
|
}
|
|
|
|
static void reverse(int[] a, int from, int to) {
|
|
while (from < --to)
|
|
swap(a, from++, to);
|
|
}
|
|
|
|
public static void rotate3(int[] a, int first, int middle, int last) {
|
|
int n = last - first;
|
|
int jump = middle - first;
|
|
int gcd = gcd(jump, n);
|
|
int cycle = n / gcd;
|
|
for (int i = 0; i < gcd; i++) {
|
|
for (int j = 0, cur = i; j < cycle - 1; j++) {
|
|
int next = cur + jump;
|
|
if (next >= n)
|
|
next -= n;
|
|
swap(a, cur, next);
|
|
cur = next;
|
|
}
|
|
}
|
|
}
|
|
|
|
static int gcd(int a, int b) {
|
|
return a == 0 ? b : gcd(b % a, a);
|
|
}
|
|
|
|
static void swap(int[] a, int i, int j) {
|
|
int t = a[j];
|
|
a[j] = a[i];
|
|
a[i] = t;
|
|
}
|
|
|
|
// random test
|
|
public static void main(String[] args) {
|
|
Random rnd = new Random(1);
|
|
for (int step = 0; step < 1000; step++) {
|
|
int n = rnd.nextInt(10) + 1;
|
|
int middle = rnd.nextInt(n);
|
|
int[] a = rnd.ints(n, 0, 10).toArray();
|
|
int[] b1 = a.clone();
|
|
rotate1(b1, 0, middle, n);
|
|
int[] b2 = a.clone();
|
|
rotate2(b2, 0, middle, n);
|
|
int[] b3 = a.clone();
|
|
rotate3(b3, 0, middle, n);
|
|
if (!Arrays.equals(b1, b2) || !Arrays.equals(b1, b3))
|
|
throw new RuntimeException();
|
|
}
|
|
}
|
|
}
|