[Silver II] 헌내기는 친구가 필요해 - 21736

문제 링크

성능 요약

메모리: 46020 KB, 시간: 296 ms

분류

너비 우선 탐색, 깊이 우선 탐색, 그래프 이론, 그래프 탐색

제출 일자

2024년 3월 4일 10:58:35

문제 설명

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

도연이가 다니는 대학의 캠퍼스는 N x 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를 출력한다.

코드

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

public class Main {
	static int count=0; //만난 사람 수
	static char[][] map; //맵
	static boolean[][] visited; //방문 여부
	static int n; //맵 크기
	static int m; //맵 크기
	static int[] dx= {-1,1,0,0}; //동서남북 이동
	static int[] dy= {0,0,-1,1}; //동서남북 이동
	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());
		
		visited=new boolean[n][m];
		map=new char[n][m];
		
		for(int i=0; i<n; i++) {
			String temp=br.readLine();
			map[i]=temp.toCharArray(); 
            //맵에 char 형태로 하나씩 저장하기
		}
		int start_x=0; int start_y=0;
		for(int i=0; i<n;i++) {
			for(int j=0;j<m;j++) {
				if(map[i][j]=='I') { //시작점 찾기
					start_x=j; 
					start_y=i;
				}
			}
		}
		dfs(start_x,start_y); //dfs 실행
		if(count==0) {
			System.out.println("TT");
		}
		else {
			System.out.println(count);
		}
		
	}
	public static void dfs(int x,int y) {
		for(int i=0; i<4; i++) {
			int nx=x+dx[i]; //동서남북 이동
			int ny=y+dy[i]; //동서남북 이동
			if(nx<0 || ny<0|| nx>=m|| ny>=n) { 
            //맵 크기를 벗어날 경우
				continue;
			}
			if(map[ny][nx]=='X' || visited[ny][nx]) { 
            //갈 수 없는 장소
				continue;
			}
			if(!visited[ny][nx] && map[ny][nx]=='O') { 
            //만약 갈 수 있지만 O라면?
				visited[ny][nx]=true;
				dfs(nx,ny);
			}
			if(!visited[ny][nx] && map[ny][nx]=='P') { 
            //갈 수 있으며 P인 경우
				visited[ny][nx]=true;
				count++;
				dfs(nx,ny);
			}
		}
		return;
	}
}

DFS 알고리즘을 이용해서 문제를 해결했다.

• 입력받은 정보로 맵을 채운다.
• I가 어디에 있는지 찾아서 시작점 x y를 설정한다.
• dfs를 실행해 현재 위치에서 동서남북으로 이동시킨다.
• 이동할 때마다 맵을 벗어나지 않는 이상 방문 여부를 체크한다.
• 만약 사람과 만났을 경우 count + 1을 해준다.
• 최종적으로 나온 count가 0이 아닐 경우 count를 출력하고, 0이면 TT를 출력한다.