[백준] S2 21736번 헌내기는 친구가 필요해 (JAVA)

문제 출처 - SW Expert Academy

문제는 여기

SW Expert Academy

[문제] 

2020년에 입학한 헌내기 도연이가 있다. 도연이는 비대면 수업 때문에 학교에 가지 못해 학교에 아는 친구가 없었다. 드디어 대면 수업을 하게 된 도연이는 어서 캠퍼스 내의 사람들과 친해지고 싶다.

도연이가 다니는 대학의 캠퍼스는 N×M 크기이며 캠퍼스에서 이동하는 방법은 벽이 아닌 상하좌우로 이동하는 것이다. 예를 들어, 도연이가 (x, y)에 있다면 이동할 수 있는 곳은 (x+1, y), (x, y+1), (x−1, y), (x, y−1)이다. 단, 캠퍼스의 밖으로 이동할 수는 없다.

불쌍한 도연이를 위하여 캠퍼스에서 도연이가 만날 수 있는 사람의 수를 출력하는 프로그램을 작성해보자.

[입력]

첫째 줄에는 캠퍼스의 크기를 나타내는 두 정수 N (1≤N≤600), M (1≤M≤600)이 주어진다.

둘째 줄부터 N개의 줄에는 캠퍼스의 정보들이 주어진다. O는 빈 공간, X는 벽, I는 도연이, P는 사람이다. I가 한 번만 주어짐이 보장된다.

[출력]

첫째 줄에 도연이가 만날 수 있는 사람의 수를 출력한다. 단, 아무도 만나지 못한 경우 TT를 출력한다.

 

---

[풀이]

1. 값을 입력받아준다.

2. 입력받으면서 도연이의 좌표값을 따로 저장해준다.

3. 도연이의 좌표를 기준으로 BFS 탐색을 진행한다.

4. BFS 탐색을 하면서 사방 탐색 중 사람을 만나면 사람의 수를 증가시켜준다.

5. BFS 탐색을 통해 구한 사람의 수를 출력해준다.

[접근]

1. BFS 탐색을 하면서 만난 사람의 수를 출력하면 되겠다고 생각하였다.

[코드]

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

public class Main {
	static int n, m;
	static char[][] map;
	static boolean[][] visited;
	static int dx, dy;
	static int cnt;
	
	static int[] dr = {0, 0, -1, 1};
	static int[] dc = {-1, 1, 0, 0};
	
	// 위치를 나타낼 point 클래스
	static class point{
		int x;
		int y;
		
		public point (int x, int y) {
			this.x = x;
			this.y = y;
		}
	}

	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		StringTokenizer st = new StringTokenizer(br.readLine());
		
		n = Integer.parseInt(st.nextToken());
		m = Integer.parseInt(st.nextToken());
		
		map = new char[n][m];
		visited = new boolean[n][m];
		
		// 입력받기
		for (int i = 0; i < n; i++) {
			String str = br.readLine();
			
			for (int j = 0; j < m; j++) {
				map[i][j] = str.charAt(j);
				
				// 도연이의 위치 저장
				if (map[i][j] == 'I') {
					dx = i;
					dy = j;
				}
			}
		}
		
		// 도연이의 위치에서 bfs 탐색
		bfs(dx, dy);
		
		// 만난 사람의 수가 0이면 TT 아니면 만난 사람의 수 출력
		if (cnt != 0)
			System.out.println(cnt);
		else
			System.out.println("TT");
	}
	
	public static void bfs(int x, int y) {
		Queue<point> q = new LinkedList<>();
		q.add(new point(x, y));
		visited[x][y] = true;
		
		while (!q.isEmpty()) {
			point pos = q.poll();
			
			for (int i = 0; i < 4; i++) {
				int nx = pos.x + dr[i];
				int ny = pos.y + dc[i];
				
				// 범위, 방문, 벽인 경우
				if (nx < 0 || ny < 0 || nx >= n || ny >= m || visited[nx][ny] || map[nx][ny] == 'X')
					continue;
				
				// 사람을 만났으면
				if (map[nx][ny] == 'P')
					cnt++; // 만난 사람 수 증가
				
				// 방문처리
				visited[nx][ny] = true;
				// 추가 탐색
				q.add(new point(nx, ny));
			}
		}
	}
}