[BOJ] 3190 뱀 - Java

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

3190번: 뱀

---

1. 문제

NxN 크기의 맵 위에서 뱀이 이동한다.

처음에 뱀의 길이는 1, 좌표는 1, 1, 방향은 오른쪽을 향한다.

뱀은 매 초마다 다음과 같은 규칙을 따른다.

1. 머리를 다음칸에 위치시킨다.

2. 위치가 벽이거나 몸일 경우 종료된다.

3. 머리가 위치한 칸에 사과가 있다면 꼬리는 움직이지 않고,
    사과가 없다면 꼬리가 있는 칸을 비워준다. 즉, 몸 길이는 변하지 않는다.

 

게임이 몇 초만에 끝나는지를 출력하라.

 

2. 풀이

골드4

시뮬레이션 문제이다.

개인적으로 좋아하는 자료구조인 큐를 2개 사용하였다.

몸의 좌표를 저장할 bodyQ, 변화할 시간과 방향을 저장할 dirQ를 사용한다.

방향 전환은 dy, dx를 사용하였다.

그리고 map 값이 빈 칸일 때, 사과 일 때, 몸일 때를 구분할 때 가독성을 높이기 위해 static final로 선언하여 사용하였다.

 

y, x 둘 다 시작은 1, 방향은 0으로 시작하였다.

map에 값을 넣어주고, bodyQ에 시작지점 좌표를 넣어준 후 출발한다.

 

1. 이동

 while문이 돌 때마다 sec는 1씩 늘어난다.

1) 먼저 이동할 좌표의 유효성 검사를 진행한다.

     - 벽이면 종료 (범위를 넘어갈 경우)

     - 몸이 있으면 종료

2) 꼬리 삭제 여부를 판단한다.

     - 사과가 있으면 그대로 둔다.

     - 빈 칸이면 꼬리를 삭제한다.

이동할 때마다 신경써야 하는 것은

bodyQ와 map 값이다.

 

2. 방향 설정

이동이 끝나면 setDir()를 호출하여 방향을 설정해준다.

현재 시간을 측정하고 dirQ의 시간과 일치할 경우 (오름차순 입력이 보장됨)

dirQ의 방향에 맞게 현재 방향이 갱신된다.

D일 경우 오른쪽으로 90도, L일 경우 왼쪽으로 90도이다.

 

이동, 방향설정이 모두 끝났으니 결국에는 유효성 검사에 의해 이동이 끝나게 되고, 그 때의 sec를 출력해준다.

 

3. 코드

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

public class Main {

    static int N, K, L, sec, y, x, d;
    static int[][] map;
    static Queue<Loc> bodyQ = new ArrayDeque<> ();
    static Queue<Dir> dirQ = new ArrayDeque<> ();

    static int[] dy = {0,1,0,-1};
    static int[] dx = {1,0,-1,0};

    static final int BLANK = 0;
    static final int APPLE = 1;
    static final int BODY = 2;

    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = null;

        N = Integer.parseInt(br.readLine());
        map = new int[N+1][N+1];
        K = Integer.parseInt(br.readLine());
        for (int i=0; i<K; i++) {
            st = new StringTokenizer(br.readLine());
            int appleY = Integer.parseInt(st.nextToken());
            int appleX = Integer.parseInt(st.nextToken());
            map[appleY][appleX] = APPLE;
        }
        L = Integer.parseInt(br.readLine());
        for (int i=0; i<L; i++) {
            st = new StringTokenizer(br.readLine());
            int second = Integer.parseInt(st.nextToken());
            String direction = st.nextToken();
            dirQ.offer(new Dir(second, direction));
        }
        // 입력 끝

        y = 1;
        x = 1;
        d = 0;
        map[y][x] = BODY;
        bodyQ.offer(new Loc(y, x));

        move();

        System.out.println(sec);
    }

    static void move() {
        while (true) {
            sec++;
            int ny = y+dy[d];
            int nx = x+dx[d];
            if (ny<=0 || ny>N || nx<=0 || nx>N) break;
            if (map[ny][nx] == BODY) break;
            if (map[ny][nx] == BLANK && !bodyQ.isEmpty()) {
                Loc tail = bodyQ.poll();
                map[tail.y][tail.x] = BLANK;
            }
            y = ny;
            x = nx;
            map[y][x] = BODY;
            bodyQ.offer(new Loc(y, x));

            // 방향설정
            setDir();
        }
    }

    static void setDir() {
        if (dirQ.isEmpty()) return;
        if (sec == dirQ.peek().second) {
            Dir dir = dirQ.poll();
            if ("D".equals(dir.direction)) {
                d = (d+1)%4;
            }
            else if ("L".equals(dir.direction)) {
                d--;
                if (d<0) d = 3;
            }
        }
    }

    static class Loc {
        int y, x;
        Loc(int y, int x) {
            this.y = y;
            this.x = x;
        }
    }

    static class Dir {
        int second;
        String direction;
        Dir (int second, String direction) {
            this.second = second;
            this.direction = direction;
        }
    }
}

 

4. 사담

시간가는줄 모르고 재밌게 풀었다.

싸피할 때는 코드의 가독성, 클린코드는 뒷전이고 어떻게든 답만 나오게 하면 된다! 였으나

현재는 문제를 풀면서 스스로 가독성을 신경쓰면서 짜고 있다고 느꼈다. (그렇다고 가독성이 좋아진건 아님)

 

물론 효율적이지 않을 수도 있겠지만, 확실히 이전보다는 가독성이 나아져서 그런지

문제를 풀면서 내가 헷갈려서 다시 코드를 왔다갔다 하는 번거로움이 줄어든 것 같다.

다른 문제들을 더 많이 풀어봐야 정말로 그런지 생각해볼 수 있을 것 같다.