-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy path1806.py
42 lines (33 loc) · 1.21 KB
/
1806.py
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
31
32
33
34
35
36
37
38
39
40
41
42
class Solution:
# 暴力解法
# 第一个for循环先把perm打乱
# 然后暴力的利用 for循环对打乱的perm数组进行恢复 看需要多少步才能回到初始的数组 step来记录需要多少步
def reinitializePermutation(self, n: int) -> int:
perm = [i for i in range(n)]
arr = perm.copy()
for i in range(n):
if (i % 2 == 0):
arr[i] = perm[int(i / 2)]
else:
arr[i] = perm[int(n / 2 + (i - 1) / 2)]
perm = arr.copy()
step = 1 # 记录需要多少步 打乱已经用了一步
while (True):
if (perm == [i for i in range(n)]): # 如果和初始数组一样,那么就退出恢复数组的for循环
break
for i in range(n):
if (i % 2 == 0):
arr[i] = perm[int(i / 2)]
else:
arr[i] = perm[int(n / 2 + (i - 1) / 2)]
perm = arr.copy()
step += 1
return step
# 模拟推理
class Solution:
# 非暴力解法
def reinitializePermutation(self, n: int) -> int:
perm = [i for i in range(n)]
arr = perm.copy()
return step
# 数论