1806. 还原排列的最少操作步数
难度 中等
给你一个偶数 n
,已知存在一个长度为 n
的排列 perm
,其中 perm[i] == i
(下标 从 0 开始 计数)。
一步操作中,你将创建一个新数组 arr
,对于每个 i
:
- 如果
i % 2 == 0
,那么 arr[i] = perm[i / 2]
- 如果
i % 2 == 1
,那么 arr[i] = perm[n / 2 + (i - 1) / 2]
然后将 arr
赋值给 perm
。
要想使 perm
回到排列初始值,至少需要执行多少步操作?返回最小的 非零 操作步数。
示例 1:
输入:n = 2
输出:1
解释:最初,perm = [0,1]
第 1 步操作后,perm = [0,1]
所以,仅需执行 1 步操作
示例 2:
输入:n = 4
输出:2
解释:最初,perm = [0,1,2,3]
第 1 步操作后,perm = [0,2,1,3]
第 2 步操作后,perm = [0,1,2,3]
所以,仅需执行 2 步操作
示例 3:
输入:n = 6
输出:4
提示:
-
2 <= n <= 1000
-
n
是一个偶数
Solution
impl Solution {
pub fn reinitialize_permutation(n: i32) -> i32 {
// 考察1的位置
// 经过找规律可以发现答案就是满足2^(res-1)=k(n-1)+1的最小res
// 但是这个res可能太大了,大到几百
// 所以只能用同余做
let mut cur = 1;
let mut res = 0;
if n == 2 {
return 1;
}
while true {
cur *= 2;
res += 1;
if cur % (n-1) == 1 {
return res;
}
cur = cur%(n-1);
}
114514
}
}
当然,也可以直接模拟这个数组操作,这个复杂度是允许的。