728x90
반응형
게임 개발
난이도 : 中 풀이 시간 : 40분
시간 제한 : 1초 메모리 제한 : 128 MB
해답
n, m = map(int, input().split())
x, y, direction = map(int, input().split())
array = []
for i in range(n):
array.append(list(map(int, input().split())))
darray = [[0] * m for _ in range(n)]
dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]
darray[x][y] = 1
count = 1
turned_count = 0
while True:
direction = direction - 1
if direction == -1:
direction = 3
nx = x + dx[direction]
ny = y + dy[direction]
if darray[nx][ny] == 0 and array[nx][ny] == 0:
darray[nx][ny] = 1
count = count + 1
x, y = nx, ny
turned_count = 0
continue
else:
turned_count = turned_count + 1
if turned_count == 4:
nx = x - dx[direction]
ny = y - dy[direction]
if array[nx][ny] == 0:
x, y = nx, ny
else:
break
turned_count = 0
print(count)
예시
# N, M을 공백을 기준으로 구분하여 입력받기
n, m = map(int, input().split())
# 방문한 위치를 저장하기 위한 맵을 생성하여 0으로 초기화
d = [[0] * m for _ in range(n)]
# 현재 캐릭터의 X 좌표, Y 좌표, 방향을 입력받기
x, y, direction = map(int, input().split())
d[x][y] = 1 # 현재 좌표 방문 처리
# 전체 맵 정보를 입력받기
array = []
for i in range(n):
array.append(list(map(int, input().split())))
# 북, 동, 남, 서 방향 정의
dx = [-1, 0, 1, 0]
dy = [0, 1, 0, -1]
# 왼쪽으로 회전
def turn_left():
global direction
direction -= 1
if direction == -1:
direction = 3
# 시뮬레이션 시작
count = 1
turn_time = 0
while True:
# 왼쪽으로 회전
turn_left()
nx = x + dx[direction]
ny = y + dy[direction]
# 회전한 이후 정면에 가보지 않은 칸이 존재하는 경우 이동
if d[nx][ny] == 0 and array[nx][ny] == 0:
d[nx][ny] = 1
x = nx
y = ny
count += 1
turn_time = 0
continue
# 회전한 이후 정면에 가보지 않은 칸이 없거나 바다인 경우
else:
turn_time += 1
# 네 방향 모두 갈 수 없는 경우
if turn_time == 4:
nx = x - dx[direction]
ny = y - dy[direction]
# 뒤로 갈 수 있다면 이동하기
if array[nx][ny] == 0:
x = nx
y = ny
# 뒤가 바다로 막혀있는 경우
else:
break
turn_time = 0
# 정답 출력
print(count)
해설
방향을 설정해서 이동하는 문제 유형에서는 dx, dy라는 별도의 리스트를 만들어 방향을 정하는 것이 효과적
728x90
반응형
'알고리즘 (Python) > 이것이 코딩 테스트다 with 파이썬' 카테고리의 다른 글
[DFS/BFS 알고리즘] 탐색 알고리즘 DFS/BFS (0) | 2021.05.27 |
---|---|
[DFS/BFS 알고리즘] 꼭 필요한 자료구조 기초 (0) | 2021.05.27 |
[구현 알고리즘] 왕실의 나이트 - 파이썬(python) (0) | 2021.05.25 |
[구현 알고리즘] 아이디어를 코드로 바꾸는 구현 (0) | 2021.05.25 |
[그리디 알고리즘] 1이 될 때까지 - 파이썬(python) (0) | 2021.05.24 |