세로 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를 밟지 못한다.
이것만 신경써서 풀어주면 쉬운 문제였다.
'알고리즘 > 백준' 카테고리의 다른 글
[백준/1260] - DFS와 BFS(JAVA) (0) | 2023.09.17 |
---|---|
[백준/1526] - 내리막길(JAVA) (0) | 2023.09.15 |
[백준/11725] - 트리의 부모 찾기(JAVA) (0) | 2023.09.15 |
[백준/2468] - 안전 영역(JAVA) (0) | 2023.09.14 |
[백준/4963] - 섬의 개수(JAVA) (2) | 2023.09.14 |