[백준]3190번: 뱀

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

3190번: 뱀

---

단순 구현 문제지만 시간이 아주 오래 걸렸다.🥲

 

이 문제 풀 때 조심할 점은,

이동 먼저, 방향 전환 나중이라는 점이다!!

나는 처음에 방향 전환 먼저 하고 이동을 나중에 해서 오답 판정을 받았었다.

 

---

 

제출 코드

빈 자리는 0, 사과는 1, 뱀의 몸통은 2로 표시하였다.

뱀의 몸통 좌표를 차례대로 큐에 담아 꼬리를 줄일 경우 꺼내서 사용하였다.

import sys
from collections import deque
input = sys.stdin.readline

# n * n 배열을 모두 0으로 초기화
n = int(input())
board = [[0]*n for _ in range(n)]

# 사과가 있는 자리는 모두 1로 표시
k = int(input())
for k in range(k):
  x, y = map(int,input().split())
  board[x-1][y-1] = 1

# 방향 변경 정보를 입력받아 저장
direction = deque()
l = int(input())
for i in range(l):
  t, d = input().split()
  direction.append((t,d))

# 다음 변경 시간과, 변경 방향
change_t, change_d = direction.popleft()


# 위, 왼, 아래, 오
dx = [-1,0,1,0]
dy = [0,-1,0,1]

# 경과시간은 0, 현재방향은 오른쪽으로 초기화
time = 0
direct = 3

# 뱀의 몸이 지나간 좌표를 차례대로 저장
# 꼬리 길이를 줄여야할 때 사용하기 위함
snake = deque()
snake.append((0,0))

x, y = 0, 0
while True:

  # 경과시간을 증가시켜준다.
  time += 1

  # 뱀이 바라보고 있는 좌표로 이동
  nx, ny = x + dx[direct], y + dy[direct]

  # 리스트 범위 벗어나는지 체크
  if 0<=nx<n and 0<=ny<n:
    # 사과가 있으면, 꼬리를 줄이지 않고 이동한다.
    if board[nx][ny] == 1:
      board[nx][ny] = 2
      snake.append((nx,ny))
      x, y = nx, ny
    # 아무것도 없으면, 꼬리를 줄이고 이동한다.
    elif board[nx][ny] == 0:
      tx, ty = snake.popleft()
      board[tx][ty] = 0 # 꼬리가 있던 자리를 0으로
      board[nx][ny] = 2
      snake.append((nx,ny))
      x, y = nx, ny
    # 이동하려는 자리에 자신의 몸이 있으면 종료한다
    elif board[nx][ny] > 1:
      break
  # 이동하려고 하는 좌표가 범위를 벗어나면 종료한다.
  else:
    break

  # 방향을 변경해야 한다면, 변경한다.
  if time == int(change_t):
    if change_d == 'D':
      if direct > 0:
        direct -= 1
      else:
        direct = 3
    elif change_d == 'L':
      if direct < 3:
        direct += 1
      else:
        direct =0
    if len(direction) > 0:
      change_t, change_d = direction.popleft()
    else:
      change_t = 0

# 결과 출력
print(time)

 

 

보완할 점

방향 변경 부분에서 무지성 if-else문 사용하는 것보다

아래처럼 하는 게 더 보기 좋고 깔끔한듯하다.

  if time == int(change_t):
    if change_d == 'D':
      direct = (direct - 1) % 4
    elif change_d == 'L':
      direct = (direct + 1) % 4