2020-01-31

import numpy as np

room_map = np.zeros((70, 70))
count_line = 0

def clean_room(room_map, j, i):
if room_map[j + 1, i] == 2:
room_map[j + 1, i] = 4
clean_room(room_map, j + 1, i)
if room_map[j - 1, i] == 2:
room_map[j - 1, i] = 4
clean_room(room_map, j - 1, i)
if room_map[j, i + 1] == 2:
room_map[j, i + 1] = 4
clean_room(room_map, j, i + 1)
if room_map[j, i - 1] == 2:
room_map[j, i - 1] = 4
clean_room(room_map, j, i - 1)
return room_map

while True:
one_line = input()
count_line = count_line+1
if not one_line:
break
else:
for i in range(len(one_line)):
if one_line[i] == ‘#’:
room_map[count_line, i] = 1
elif one_line[i] == ‘.’:
room_map[count_line, i] = 2
elif one_line[i] == ‘@’:
room_map[count_line, i] = 3
start_i=i
start_j=count_line
else:
print(‘Input error’)
break

room_map[start_j, start_i] = 4
clean_room(room_map, start_j, start_i)

for j in range(0, 70):
for i in range(0, 70):
if room_map[j, i] == 4:
print(’ ‘, end="")
elif room_map[j, i] == 2:
print(’.’, end="")
elif room_map[j, i] == 1:
print(’#’, end="")
elif room_map[j, i] == 0:
break
if room_map[j, 0] != 0:
print()