Baekjoon 21736 헌내기는 친구가 필요해 JAVA

21736번: 헌내기는 친구가 필요해

문제

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$N$개의 줄에는 캠퍼스의 정보들이 주어진다. O는 빈 공간, X는 벽, I는 도연이, P는 사람이다. I가 한 번만 주어짐이 보장된다.

 

[출력]

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

풀이

해당 문제는 도연이가 만날 수 있는 사람 수를 출력하는 문제다.

전형적인 bfs, dfs 문제로 bfs를 자주 사용하기 때문에 오랜만에 dfs를 이용하여 해결하였다. 로직은 아래와 같다.

1. 입력값을 처리하면서 도연이 위치 갱신

2. 도연이 위치에서부터 dfs를 돌리면서 체크함

    2-1. 범위를 벗어나거나 방문했거나 벽이면 다음 반복 진행

    2-2. 사람이면 카운트 증가

    2-3 dfs 반복

코드

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

public class Main {

	static int N, M, sy, sx, dy[]= {-1,1,0,0}, dx[]= {0,0,-1,1}, ans;
	static char[][] map;
	static boolean[][] visit;
	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];
		visit = new boolean[N][M];
		for (int i = 0; i < N; i++) {
			map[i] = br.readLine().toCharArray();
			for (int j = 0; j < M; j++) {
				if(map[i][j] == 'I') { // 도연이 위치 저장
					sy = i;
					sx = j;
				}
			}
		}
		dfs(sy,sx);
		if(ans != 0)
			System.out.println(ans);
		else
			System.out.println("TT");
	}
	private static void dfs(int y, int x) {
		visit[y][x] = true;
		for (int i = 0; i < 4; i++) {
			int ny = y + dy[i];
			int nx = x + dx[i];
			if(ny < 0 || nx < 0 || ny >= N || nx >= M || visit[ny][nx] || map[ny][nx] =='X')
				continue; // 배열 범위 벗어나거나 방문했거나 벽인경우
			if(map[ny][nx] =='P') // 사람이면 카운트 증가
				ans++;
			dfs(ny,nx);
		}	
	}
}

 

GitHub - JUNGSOONIL/Algorithm-JAVA: 알고리즘 문제 해결 자바 소스 코드