You are given an even integer n
. You initially have a permutation perm
of size n
where perm[i] == i
(0-indexed).
In one operation, you will create a new array arr
, and for each i
:
- If
i % 2 == 0
, thenarr[i] = perm[i / 2]
. - If
i % 2 == 1
, thenarr[i] = perm[n / 2 + (i - 1) / 2]
.
You will then assign arr
to perm
.
Return the minimum non-zero number of operations you need to perform on perm
to return the permutation to its initial value.
Input: n = 2 Output: 1 Explanation: perm = [0,1] initially. After the 1st operation, perm = [0,1] So it takes only 1 operation.
Input: n = 4 Output: 2 Explanation: perm = [0,1,2,3] initially. After the 1st operation, perm = [0,2,1,3] After the 2nd operation, perm = [0,1,2,3] So it takes only 2 operations.
Input: n = 6 Output: 4
2 <= n <= 1000
n
is even.
impl Solution {
pub fn reinitialize_permutation(n: i32) -> i32 {
let n = n as usize;
let mut perm = (0..n).collect::<Vec<_>>();
for ret in 1..=n as i32 {
let mut arr = vec![0; n];
for i in 0..n {
if i % 2 == 0 {
arr[i] = perm[i / 2];
} else {
arr[i] = perm[n / 2 + (i - 1) / 2];
}
}
perm = arr;
if perm.iter().enumerate().all(|(i, &x)| i == x) {
return ret;
}
}
unreachable!()
}
}