[백준 14503번] 로봇청소기 - 파이썬

문제 링크:

https://www.acmicpc.net/problem/14503

14503번: 로봇 청소기

 

백준 로봇청소기 문제

 

---

풀이 과정:

 

이 문제는 ★전형적인 시물레이션 문제★이다.

별도의 알고리즘이 필요하기보다는 문제에서 요구하는 내용을 오류 없이 성실하게 구현만 할 수 있다면 풀 수 있다.

 

<문제 풀이를 위한 테크닉>

1. 방향을 설정해서 이동하는 문제 유형에서는 dx, dy라는 별도의 리스트를 만들어 방향을 정하는 것이 효과적이다.

2. 리스트 컴프리헨션 문법을 이용해 2차원 리스트를 초기화하면 효율적이다.

 

풀이 코드:

 

# 14503번 로봇청소기

# 세로 크기와 가로 크기 입력받기
n, m = map(int, input().split())

# 로봇청소기 이동표시 여부 지도 생성
vaccum_move_map = [[0] * m for _ in range(n)]

# 로봇 청소기가 있는 칸의 좌표와 바라보는 방향 입력받기
x, y, direction = map(int, input().split())
# 현재 좌표 이동 처리
vaccum_move_map[x][y] = 1

# 지도 입력받기
vaccum_map = []
for i in range(n):
    vaccum_map.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

# 처음 로봇청소기 위치 청소
cnt = 1
# 회전한 횟수
turn_time = 0

# 시물레이션 시작
while True:   
  # 왼쪽으로 회전
  turn_left()
  # 이동할려는 로봇청소기의 x, y 좌표
  nx = x + dx[direction]
  ny = y + dy[direction]
  # 로봇청소기가 이동할려는 칸이 청소하지 않았으며 빈칸인 경우
  if vaccum_move_map[nx][ny] == 0 and vaccum_map[nx][ny] == 0:
    # 이동할려는 칸을 청소
    vaccum_move_map[nx][ny] = 1
    # 청소한 칸 추가
    cnt += 1
    # 로봇청소기 위치 이동
    x, y = nx, ny
    # 회전 횟수 초기화
    turn_time = 0
    continue
    # 그 외의 경우 회전한 횟수 1 증가
  else:
    turn_time += 1
  # 네 방향 모두 돌았을 때 청소할 구역이 없는 경우
  if turn_time == 4:
    # 후진하는 방향의 x, y 좌표
    nx = x - dx[direction]
    ny = y - dy[direction]
    # 뒤쪽 방향으로 이동이 가능하다면
    if vaccum_map[nx][ny] == 0:
      x, y = nx, ny
    # 그렇지 않다면
    else:
      break
    turn_time = 0
      
# 정답 출력
print(cnt)

 

 

14503번 로봇청소기