[python] 백준3190_뱀 (구현)

문제

 

풀이

이차원 리스트상의 맵에서 뱀이 이동하도록 리스트의 특정 위치에서 동,남,서,북의 위치로 이동하는 기능을 구현해야 한다. 또한, 뱀의 머리가 뱀의 몸에 닿는 순간 종료해야 하므로, 매 시점마다 뱀이 존재하는 위치를 항상 이차원 리스트에 기록해야 한다.

n = int(input())
k = int(input())
data= [[0]*(n+1) for _ in range(n+1)]
info = []

for _ in range(k):
    a,b = map(int,input().split())
    data[a][b] = 1

l = int(input())
for _ in range(l):
    x,c = input().split()
    info.append((int(x),c))

# 동,남,서,북
dx = [0,1,0,-1] 
dy = [1,0,-1,0]

def turn(direction, c):
    if c=='L':
        direction = (direction-1)%4
    else :
        direction = (direction+1)%4
    return direction

def simulate():
    x,y = 1,1 # 처음 뱀의 위치
    data[x][y]=2 # 뱀이 있는 곳은 2로
    direction = 0 # 처음 동쪽
    time = 0
    index = 0 # 다음에 회전할 정보
    q = [(x,y)] # 뱀이 차지하고 있는 위치 정보(꼬리가 앞쪽)
    
    while True:
        nx = x + dx[direction]
        ny = y + dy[direction]
        # 맵의 범위 안족이고 뱀의 몸통이 없는 곳이면
        if 1 <= nx and nx <=n and 1<=ny and ny <=n and data[nx][ny] !=2 :
            # 사과가 없다면 이동 후에 꼬리 제거
            if data[nx][ny] == 0:
                data[nx][ny] = 2
                q.append((nx,ny))
                px, py = q.pop(0)
                data[px][py] = 0
            # 사과가 있다면 이동 후에 꼬리 그대로
            if data[nx][ny] == 1:
                data[nx][ny] =2
                q.append((nx,ny))
        
        # 벽이나 몸통에 부딪혔다면
        else:
            time +=1
            break
        x,y = nx, ny # 다음 위치로 머리 이동
        time += 1
        if index < l and time == info[index][0] : # 회전할 시간인 경우 회전
            direction = turn(direction,info[index][1])
            index += 1
    return time