백준 14503 로봇청소기 -python
문제
- 시뮬레이션 문제
문제풀이
- 방향을 북동남서 방향으로 바꿔줘야 한다
- 벽인지, 빈칸인지를 잘 구별해주어야 한다
- 후진을 할 방향을 잘 선택해주면 된다
code
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
56
# 방향은 북동남서
dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]
# 방향 바꿔주기
def change(d):
if(d == 0):
return 3
elif(d == 1):
return 0
elif(d == 2):
return 1
elif(d == 3):
return 2
def find(r,c,d):
cnt = 1
x = r
y = c
arr[x][y] = 2
while(True):
dc = d
for i in range(4):
empty = 0
dc = change(dc)
nx = x + dx[dc]
ny = y + dy[dc]
# 유효 범위 안에 있고, 빈칸이라면
if(0<=nx<n and 0<=ny<m and arr[nx][ny] == 0):
cnt += 1
x = nx
y = ny
arr[nx][ny] = 2
d = dc
empty = 1
break
# 4방향 모두 탐색 후 모든 칸이 청소가 되었다면
if(empty == 0):
# 후진
if(d == 0):
x += 1
elif(d == 1):
y -= 1
elif(d == 2):
x -= 1
elif(d == 3):
y += 1
# 후진하려는 칸이 벽이라면 stop
if(arr[x][y] == 1):
break
return cnt
n, m = map(int, input().split())
r, c, d = map(int, input().split())
arr = [list(map(int, input().split())) for _ in range(n)]
res = find(r,c,d)
print(res)
