반응형
나의 풀이
n, m = map(int, input().split())
a, b, d = map(int, input().split())
world = []
for _ in range(n):
world.append(list(map(int, input().split())))
def turn_left():
global d
if d == 0:
d = 3
else:
d -= 1
def visited():
global world
world[a][b] = 2
cnt = 1
turn_cnt = 0
left_types = [[0,-1], [-1,0], [0,1], [1,0]]
back_types = [[1,0], [0,-1], [-1,0], [0,1]]
visited()
while True:
tmp_a, tmp_b = a + left_types[d][0], b + left_types[d][1]
check_point = world[tmp_a][tmp_b]
if check_point == 0:
turn_left()
a, b = tmp_a, tmp_b
visited()
cnt += 1
turn_cnt = 0
else:
turn_left()
turn_cnt += 1
if turn_cnt >= 4:
tmp_a , tmp_b = a + back_types[d][0], b + back_types[d][1]
turn_cnt = 0
if world[tmp_a][tmp_b] == 1:
break
else:
a, b = tmp_a, tmp_b
print(cnt)
답안 예시
# 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)
반응형