[Silver I] 양 - 3184

문제 링크

성능 요약

메모리: 16684 KB, 시간: 180 ms

분류

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

제출 일자

2024년 3월 26일 19:13:37

문제 설명

미키의 뒷마당에는 특정 수의 양이 있다. 그가 푹 잠든 사이에 배고픈 늑대는 마당에 들어와 양을 공격했다.

마당은 행과 열로 이루어진 직사각형 모양이다. 글자 '.' (점)은 빈 필드를 의미하며, 글자 '#'는 울타리를, 'o'는 양, 'v'는 늑대를 의미한다.

한 칸에서 수평, 수직만으로 이동하며 울타리를 지나지 않고 다른 칸으로 이동할 수 있다면, 두 칸은 같은 영역 안에 속해 있다고 한다. 마당에서 "탈출"할 수 있는 칸은 어떤 영역에도 속하지 않는다고 간주한다.

다행히 우리의 양은 늑대에게 싸움을 걸 수 있고 영역 안의 양의 수가 늑대의 수보다 많다면 이기고, 늑대를 우리에서 쫓아낸다. 그렇지 않다면 늑대가 그 지역 안의 모든 양을 먹는다.

맨 처음 모든 양과 늑대는 마당 안 영역에 존재한다.

아침이 도달했을 때 살아남은 양과 늑대의 수를 출력하는 프로그램을 작성하라.

입력

첫 줄에는 두 정수 R과 C가 주어지며(3 ≤ R, C ≤ 250), 각 수는 마당의 행과 열의 수를 의미한다.

다음 R개의 줄은 C개의 글자를 가진다. 이들은 마당의 구조(울타리, 양, 늑대의 위치)를 의미한다.

출력

하나의 줄에 아침까지 살아있는 양과 늑대의 수를 의미하는 두 정수를 출력한다.

코드 (DFS 풀이)

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

public class Main {
	static char[][] map; //정원 맵
	static boolean[][] visited; //방문 여부
	
	static int R; //가로
	static int C; //세로
	
	static int total_sheep=0; //최종 양 수
	static int total_wolves=0; //최종 늑대 수
	
	static int sheep=0; //양
	static int wolves=0; //늑대
	
	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));
		StringTokenizer st;
		st=new StringTokenizer(br.readLine());
		R=Integer.parseInt(st.nextToken());
		C=Integer.parseInt(st.nextToken()); //가로 세로 변수 입력받기
        
		map=new char[R][C]; //맵 크기 설정
		visited=new boolean[R][C];
		
		/* 
		for(int i=0; i<R; i++) {
			String temp = br.readLine();
			for(int j=0; j<C; j++) {
				map[i][j]=temp.charAt(j);
			}
		}
		*/ //밑에 for문 하나로 쓰는 것이 더 편해서 고침.
		
		for(int i=0; i<R; i++) { //맵 입력받기
			map[i]=br.readLine().toCharArray();
		}
		
		for(int i=0; i<R; i++) {
			for(int j=0; j<C; j++) {
				if((map[i][j]=='v' || map[i][j]=='o') && !visited[i][j]) {
                    //만약 맵에 늑대가 있거나 양이 있고 방문하지 않았다면?
					sheep=0; //초기화
					wolves=0;
					DFS(i,j); //DFS 실행
					if(sheep > wolves) { //만약 양이 더 많다면 늑대를 몰아냄
						total_sheep+=sheep;
					}
					else { //늑대가 더 많다면 양 없어짐.
						total_wolves+=wolves;
					}
				}
			}

		}
		System.out.println(total_sheep + " "+ total_wolves);
	}
	
	public static void DFS(int x,int y) {
		visited[x][y]=true; //방문 TRUE
		if(map[x][y]=='v') {//만약 늑대가 있다면?
			wolves++;
		}
		else if(map[x][y]=='o') {//만약 양이 있다면?
			sheep++;
		}
		for(int i=0; i<4; i++) {//동서남북으로 이동하면서 확인
			int nx=x+dx[i];
			int ny=y+dy[i];
			if(nx<0 || nx>=R || ny<0 || ny>=C) { //맵 넘어가면 CONTINUE
				continue;
			}
			if(map[nx][ny]=='#'|| visited[nx][ny]) { //울타리에 걸리거나 방문했다면 CONTINUE
				continue;
			}
			visited[nx][ny]=true; //동서남북 이동한 위치 방문 TRUE
			DFS(nx,ny); //이동한 위치에서 DFS 실행
		}
		return;
	}
}

DFS 알고리즘을 사용한 풀이다.

실행단계
1. 가로 세로를 입력받는다.
2. 맵 크기와 방문 여부 배열을 생성해주고 맵 크기를 입력받는다.
3. 만약 맵의 위치에 양이나 늑대가 있고 방문하지 않은 위치라면 늑대와 양의 수를 0으로 설정하고 DFS 알고리즘을 실행한다.
3-1. 동서남북으로 이동해가며 울타리가 아니라면 계속 이동한다.
3-2. 만약 이동한 위치에 양이나 늑대가 있다면 각각의 수 + 1을 해준다.
3-3. 이동한 위치의 방문 여부를 TRUE로 변경한다.
4-1. 만약 울타리 구역에 늑대가 양보다 많다면 늑대만 최종 늑대의 수에 더해준다.
4-2. 만약 울타리 구역에 양이 늑대보다 많다면 양만 최종 양의 수에 더해준다.
5. 최종 양 수와 늑대 수를 출력한다.