forked from keon/algorithms
-
Notifications
You must be signed in to change notification settings - Fork 0
/
garage.py
42 lines (35 loc) · 1.3 KB
/
garage.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
# There is a parking lot with only one empty spot. Given the initial state
# of the parking lot and the final state. Each step we are only allowed to
# move a car
# out of its place and move it into the empty spot.
# The goal is to find out the least movement needed to rearrange
# the parking lot from the initial state to the final state.
# Say the initial state is an array:
# [1,2,3,0,4],
# where 1,2,3,4 are different cars, and 0 is the empty spot.
# And the final state is
# [0,3,2,1,4].
# We can swap 1 with 0 in the initial array to get [0,2,3,1,4] and so on.
# Each step swap with 0 only.
# Edited by cyberking-saga
def garage(initial, final):
steps = 0
while initial != final:
zero = initial.index(0)
if zero != final.index(0):
car_to_move = final[zero]
pos = initial.index(car_to_move)
initial[zero], initial[pos] = initial[pos], initial[zero]
else:
for i in range(len(initial)):
if initial[i] != final[i]:
initial[zero], initial[i] = initial[i], initial[zero]
break
steps += 1
return steps
if __name__ == "__main__":
initial = [1, 2, 3, 0, 4]
final = [0, 3, 2, 1, 4]
print("initial:", initial)
print("final:", final)
print(garage(initial, final))