-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathfuga.py
55 lines (39 loc) · 931 Bytes
/
fuga.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
43
44
45
46
47
48
49
50
51
52
53
54
55
LIVRE = "."
SAIDA = "E"
FOGO = "F"
INICIO = "S"
PAREDE = "#"
var = [(-1,0), (1,0), (0,1), (0, -1)]
def isValido(x,y):
if ((x >= 0 or x < M) and (y >= 0 or y < M)):
return True
else:
return False
def getFogos():
aux = []
for i in range(N):
for j in range(M):
if (adj[i][j] == -1 or adj[i][j] == FOGO):
aux.append((i,j))
return aux
def dfs(x,y):
fogos = getFogos()
for i in range(len(fogos)):
x = fogos[i][0]
y = fogos[i][1]
for j in range(4):
if (isValido(x + var[j][0], y + var[j][1])):
adj[x + var[j][0]][y + var[j][1]] = -1
print adj
consultas = int(raw_input())
for i in range(consultas):
N, M = map(int, raw_input().split())
adj = [["." for x in range(M)] for y in range(N)]
for i in range(N):
linha = raw_input().split()
for j in range(len(linha)):
adj[i] = linha[j]
for i in range(N):
for j in range(M):
if (adj[i][j] == INICIO):
dfs(i,j)