2020년에 입학한 헌내기 도연이가 있다. 도연이는 비대면 수업 때문에 학교에 가지 못해 학교에 아는 친구가 없었다. 드디어 대면 수업을 하게 된 도연이는 어서 캠퍼스 내의 사람들과 친해지고 싶다.
도연이가 다니는 대학의 캠퍼스는 크기이며 캠퍼스에서 이동하는 방법은 벽이 아닌 상하좌우로 이동하는 것이다. 예를 들어, 도연이가 (, )에 있다면 이동할 수 있는 곳은 (, ), (, ), (, ), (, )이다. 단, 캠퍼스의 밖으로 이동할 수는 없다.
불쌍한 도연이를 위하여 캠퍼스에서 도연이가 만날 수 있는 사람의 수를 출력하는 프로그램을 작성해보자.
첫째 줄에는 캠퍼스의 크기를 나타내는 두 정수 ($ 1 \leq N \leq 600M 1 \leq M \leq 600$)이 주어진다.
둘째 줄부터 개의 줄에는 캠퍼스의 정보들이 주어진다. O는 빈 공간, X는 벽, I는 도연이, P는 사람이다. I가 한 번만 주어짐이 보장된다.
첫째 줄에 도연이가 만날 수 있는 사람의 수를 출력한다. 단, 아무도 만나지 못한 경우 TT를 출력한다.
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.StringTokenizer;
public class Main {
private static char[][] arr;
private static boolean[][] visited;
private static int meetPeople = 0;
private static int[] moveX = { -1, 1, 0, 0 };
private static int[] moveY = { 0, 0, -1, 1 };
private static int n, m;
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
StringTokenizer st = new StringTokenizer(br.readLine());
n = Integer.parseInt(st.nextToken());
m = Integer.parseInt(st.nextToken());
arr = new char[n][m];
visited = new boolean[n][m];
int startX = 0;
int startY = 0;
for(int i = 0; i < n; i++) {
String line = br.readLine();
for(int j = 0; j < m; j++) {
arr[i][j] = line.charAt(j);
// 도연이의 위치 찾기
if(arr[i][j] == 'I') {
startX = i;
startY = j;
}
}
}
dfs(startX, startY);
System.out.println(meetPeople == 0 ? "TT" : meetPeople);
}
public static void dfs(int x, int y) {
if(visited[x][y]) return;
visited[x][y] = true;
// 사람을 만났을때 meetPeople + 1
if(arr[x][y] == 'P') meetPeople++;
for(int i = 0; i < 4; i++) {
int dx = x + moveX[i];
int dy = y + moveY[i];
if(dx >= 0 && dy >= 0 && dx < n && dy < m && arr[dx][dy] != 'X') {
dfs(dx, dy);
}
}
}
}
해당 문제는 DFS를 이용하여 문제를 해결했습니다.
arr 배열에 입력해줍니다.startX, startY에 값을 입력해줍니다.startX, startY 위치에서 dfs를 시작합니다.P가 있으면 사람을 만난 수를 의미하는 meetPeople에 +1 해줍니다.meetPeople 값을 확인하여 0이면 TT 아니면 meetPeople 값을 출력해줍니다.