[JAVA] BOJ 백준 3190 뱀

문제 바로가기

📌 이 문제의 유의점은 반드시 정해진 순서대로 코드를 짜야 하는 것이다.
머리와 꼬리가 동시에 움직임을 결정하는 것이 아니라 머리가 움직이고! 후에 꼬리를 그대로 둘지 이동할지 결정해야 한다

전체코드

import java.io.*;
import java.util.*;

public class Main {
    static class timer{
        char dir;
        int time;
        timer(char dir, int time){
            this.dir = dir;
            this.time = time;
        }
    }
    static boolean[][] map;
    static boolean[][] snake;
    static int[] dx = {0, 1, 0, -1}, dy = {1, 0, -1, 0};
    public static void main(String[] args) throws Exception{
        BufferedReader bf = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(bf.readLine());
        map = new boolean[N][N];
        snake = new boolean[N][N];
        int K = Integer.parseInt(bf.readLine());
        for(int i=0; i<K; i++){
            StringTokenizer st = new StringTokenizer(bf.readLine(), " ");
            int x = Integer.parseInt(st.nextToken());
            int y = Integer.parseInt(st.nextToken());
            map[x-1][y-1] = true;
        }
        ArrayDeque<timer> q = new ArrayDeque<>();
        int L = Integer.parseInt(bf.readLine());
        for(int i=0; i<L; i++){
            StringTokenizer st = new StringTokenizer(bf.readLine(), " ");
            int x = Integer.parseInt(st.nextToken());
            char c = st.nextToken().charAt(0);
            q.offer(new timer(c, x));
        }
        int time = 1;
        int hx = 0; int hy = 0;
        int tx = 0; int ty = 0;
        int d = 0;
        ArrayDeque<int[]> road = new ArrayDeque<>();
        while(true){
            if(!q.isEmpty() && q.peek().time < time){
                timer t = q.poll();
                if(t.dir == 'L')
                    d = (d+3) % 4;
                else if(t.dir == 'D')
                    d = (d+1) % 4;
            }
            hx += dx[d]; hy += dy[d];

            if(hx < 0 || hx >= N || hy < 0 || hy >= N) break; // 벽에 부딪히거나
            if(snake[hx][hy]) break; // 자기자신의 몸과 부딪힘
            snake[hx][hy] = true;
            road.offer(new int[]{hx, hy});

            if(map[hx][hy]) map[hx][hy] = false;
            else {
                snake[tx][ty] = false;
                tx = road.peek()[0];
                ty = road.poll()[1];
            }
//            System.out.println(hx+", " +hy)
            time++;
        }
        System.out.println(time);
    }
}

한 줄 씩 보자!

static class timer{
    char dir;
    int time;
    timer(char dir, int time){
        this.dir = dir;
        this.time = time;
    }
}
static boolean[][] map;
static boolean[][] snake;
static int[] dx = {0, 1, 0, -1}, dy = {1, 0, -1, 0};

timer 클래스는 입력받은 위치 변환 정보를 받는 클래스이다.

map은 사과의 정보를 표시한다(true일 때, 해당 위치에 사과가 존재한다는 뜻)

snake는 뱀의 위치들을 저장한다(true이면 해당 위치에 뱀이 존재한다는 뜻)

dx, dy로 다음 위치를 지정한다.

int N = Integer.parseInt(bf.readLine());
map = new boolean[N][N];
snake = new boolean[N][N];
int K = Integer.parseInt(bf.readLine());
for(int i=0; i<K; i++){
    StringTokenizer st = new StringTokenizer(bf.readLine(), " ");
    int x = Integer.parseInt(st.nextToken());
    int y = Integer.parseInt(st.nextToken());
    map[x-1][y-1] = true;
}
ArrayDeque<timer> q = new ArrayDeque<>();
int L = Integer.parseInt(bf.readLine());
for(int i=0; i<L; i++){
    StringTokenizer st = new StringTokenizer(bf.readLine(), " ");
    int x = Integer.parseInt(st.nextToken());
    char c = st.nextToken().charAt(0);
    q.offer(new timer(c, x));
}

사과의 위치를 받아서 map에 사과 정보를 갱신한다.

위치 변경 정보를 받아서 Queue에 저장해준다.

int time = 1;
int hx = 0; int hy = 0;
int tx = 0; int ty = 0;
int d = 0;
ArrayDeque<int[]> road = new ArrayDeque<>();
while(true){
    if(!q.isEmpty() && q.peek().time < time){
        timer t = q.poll();
        if(t.dir == 'L')
            d = (d+3) % 4;
        else if(t.dir == 'D')
            d = (d+1) % 4;
    }
    hx += dx[d]; hy += dy[d];

    if(hx < 0 || hx >= N || hy < 0 || hy >= N) break; // 벽에 부딪히거나
    if(snake[hx][hy]) break; // 자기자신의 몸과 부딪힘
    snake[hx][hy] = true;
    road.offer(new int[]{hx, hy});

    if(map[hx][hy]) map[hx][hy] = false;
    else {
        snake[tx][ty] = false;
        tx = road.peek()[0];
        ty = road.poll()[1];
    }
    time++;
}
System.out.println(time);
}

time 변수에 지난 시간을 저장한다. hx, hy는 뱀의 머리 위치를 tx, ty는 뱀의 꼬리 위치를 저장한다.

d에는 이동 방향을 저장한다.

Queue에 머리의 이동 위치를 저장하여 꼬리가 따라서 이동할 수 있게 한다.

먼저 이동해야 되는 타임에 이동 방향을 변경해준다.

이때 머리가 벽에 부딪히거나 몸통과 만나게 되면 게임이 종료되므로 break 처리를 한다.

그 뒤 꼬리의 이동을 설정해준다.

마지막으로 시간을 증가시킨다.