세로 R칸, 가로 C칸으로 된 표 모양의 보드가 있다. 보드의 각 칸에는 대문자 알파벳이 하나씩 적혀 있고, 좌측 상단 칸 (1행 1열) 에는 말이 놓여 있다.

말은 상하좌우로 인접한 네 칸 중의 한 칸으로 이동할 수 있는데, 새로 이동한 칸에 적혀 있는 알파벳은 지금까지 지나온 모든 칸에 적혀 있는 알파벳과는 달라야 한다. 즉, 같은 알파벳이 적힌 칸을 두 번 지날 수 없다.

좌측 상단에서 시작해서, 말이 최대한 몇 칸을 지날 수 있는지를 구하는 프로그램을 작성하시오. 말이 지나는 칸은 좌측 상단의 칸도 포함된다.

입력


첫째 줄에 R과 C가 빈칸을 사이에 두고 주어진다. (1 ≤ R,C ≤ 20) 둘째 줄부터 R개의 줄에 걸쳐서 보드에 적혀 있는 C개의 대문자 알파벳들이 빈칸 없이 주어진다.

 

출력


첫째 줄에 말이 지날 수 있는 최대의 칸 수를 출력한다.

 

풀이


import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;

public class Main {

    static StringTokenizer st;
    static char[][] field;
    static boolean[] visited;
    static int R, C, count, max;

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

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        st = new StringTokenizer(br.readLine());
        R = Integer.parseInt(st.nextToken());
        C = Integer.parseInt(st.nextToken());
        field = new char[R][C];
        for (int i=0; i<R; i++){
            String row = br.readLine();
            for (int j=0; j<C; j++){
                field[i][j] = row.charAt(j);
            }
        }

        //DFS
        max = 1;
        visited = new boolean[26];
        count = 1;
        DFS(0,0,count);
        System.out.println(max);
    }
    static void DFS(int x, int y, int count) {
        visited[field[x][y]-65] = true;
        max = Math.max(max, count);
        for (int i=0; i<4; i++){
            int nx = x + dx[i];
            int ny = y + dy[i];
            if (nx >= 0 && ny >= 0 && nx < R && ny < C){
                if (!visited[field[nx][ny]-65]){
                    DFS(nx, ny, count+1);
                }
            }
        }
        visited[field[x][y]-65] = false;
    }
}

해설


'방문한 배열에 대해 어떻게 초기화를 적절히 해 줄 수 있을까'라는 구현이 가장 핵심인 DFS문제였다.

개인적으로 DP를 왜 사용하는 지에 대한 이유를 알 수 있는 문제라 중요하다고 생각했다.

[0,0]에서 [1,0]으로 이동할때와 [0,1]로 이동할때의 최댓값에 따라 이동하는데, visited 배열을 가만히 놔두면

기존 움직임에 대해서 다음 DFS가 체크할 수 없다.

static StringTokenizer st;
static char[][] field;
static boolean[] visited;
static int R, C, count, max;
static int[] dx = {0,0,-1,1};
static int[] dy = {1,-1,0,0};

방향을 체크해야 하니 델타배열을 생성하고, field와 visited를 만든다.

처음에는 visited와 alphabet이라는 배열을 두개 생성해서 visited와 alphabet에 대해 따로따로 체크했다.

그러나 alphabet 배열을 제거했고 이유는 다음과 같았다.

배열을 두 개 만든 이유는 방문했던 위치와 알파벳이 별개로 돌아갈거라고 생각했는데 그렇지 않았다

ㅇ예를 들어 다음 입력을 가정해보자.

2 4
CAAB
ACAB

C에서 [0,1]인 A로 이동했을 때, [1,1]인 C는 가면 안된다.

그러나 방문하지 않았던 배열이기 때문에 visited와 alphabet을 따로 만들었었는데

생각해보면 '방문했던 배열의 알파벳은 다시 밟지 않는다'의 원칙이 있기 때문에 visited가 곧 alphabet이 된다.

다만 visited배열을 알파벳 대문자와 같은 26개의 크기로 만들어주고, 밟을 때마다 알파벳에 체크를 해준다.

public class Main {

    static StringTokenizer st;
    static char[][] field;
    static boolean[] visited;
    static int R, C, count, max;

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

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        st = new StringTokenizer(br.readLine());
        R = Integer.parseInt(st.nextToken());
        C = Integer.parseInt(st.nextToken());
        field = new char[R][C];
        for (int i=0; i<R; i++){
            String row = br.readLine();
            for (int j=0; j<C; j++){
                field[i][j] = row.charAt(j);
            }
        }

        //DFS
        max = 1;
        visited = new boolean[26];
        count = 1;
        DFS(0,0,count);
        System.out.println(max);
    }

그래서 배열을 초기화하고 field에 값을 채운 뒤 DFS를 돌려준다.

    static void DFS(int x, int y, int count) {
        visited[field[x][y]-65] = true;
        max = Math.max(max, count);
        for (int i=0; i<4; i++){
            int nx = x + dx[i];
            int ny = y + dy[i];
            if (nx >= 0 && ny >= 0 && nx < R && ny < C){
                if (!visited[field[nx][ny]-65]){
                    DFS(nx, ny, count+1);
                }
            }
        }
        visited[field[x][y]-65] = false;
    }

가장 중요한 DFS 함수이다.

맨 마지막 줄에 visited를 false로 돌려놔야하는게 핵심이다.

아까의 input처럼 처음 C를 밟고 [1,0]에 있는 A를 밟았을때 visited를 체크하고 이걸 돌려놓지 않으면

[0,1]을 실행할 때 A를 밟지 못한다.

이것만 신경써서 풀어주면 쉬운 문제였다.

+ Recent posts