문제
미키의 뒷마당에는 특정 수의 양이 있다. 그가 푹 잠든 사이에 배고픈 늑대는 마당에 들어와 양을 공격했다.
마당은 행과 열로 이루어진 직사각형 모양이다. 글자 '.' (점)은 빈 필드를 의미하며, 글자 '#'는 울타리를, 'o'는 양, 'v'는 늑대를 의미한다.
한 칸에서 수평, 수직만으로 이동하며 울타리를 지나지 않고 다른 칸으로 이동할 수 있다면, 두 칸은 같은 영역 안에 속해 있다고 한다. 마당에서 "탈출"할 수 있는 칸은 어떤 영역에도 속하지 않는다고 간주한다.
다행히 우리의 양은 늑대에게 싸움을 걸 수 있고 영역 안의 양의 수가 늑대의 수보다 많다면 이기고, 늑대를 우리에서 쫓아낸다. 그렇지 않다면 늑대가 그 지역 안의 모든 양을 먹는다.
맨 처음 모든 양과 늑대는 마당 안 영역에 존재한다.
아침이 도달했을 때 살아남은 양과 늑대의 수를 출력하는 프로그램을 작성하라.
입력
첫 줄에는 두 정수 R과 C가 주어지며(3 ≤ R, C ≤ 250), 각 수는 마당의 행과 열의 수를 의미한다.
다음 R개의 줄은 C개의 글자를 가진다. 이들은 마당의 구조(울타리, 양, 늑대의 위치)를 의미한다.
출력
하나의 줄에 아침까지 살아있는 양과 늑대의 수를 의미하는 두 정수를 출력한다.
예제 입력 1 복사
6 6
...#..
.##v#.
#v.#.#
#.o#.#
.###.#
...###
예제 출력 1 복사
0 2
예제 입력 2 복사
8 8
.######.
#..o...#
#.####.#
#.#v.#.#
#.#.o#o#
#o.##..#
#.v..v.#
.######.
예제 출력 2 복사
3 1
예제 입력 3 복사
9 12
.###.#####..
#.oo#...#v#.
#..o#.#.#.#.
#..##o#...#.
#.#v#o###.#.
#..#v#....#.
#...v#v####.
.####.#vv.o#
.......####.
예제 출력 3 복사
3 5
풀이
solution에서 dfs를 호출할때마다 그 호출된 영역에서 살아남은 양과 늑대의 수를 계산해서 전체 양, 늑대의 수에 더해주는 식으로 풀었다
import java.io.*;
import java.util.*;
public class Main {
static int N, M, lamb, wolf, res_lamb = 0, res_wolf = 0;
static StringTokenizer st;
static String s;
static int[][] dir = {{1,0},{0,1},{-1,0},{0,-1}};
static char[][] arr;
static boolean[][] visited;
static void solution(){
for (int i = 0; i < N; i++){
for (int j = 0; j < M; j++){
lamb = 0;
wolf = 0;
//울타리가 아니면 dfs 시작
if (!visited[i][j] && arr[i][j] != '#'){
dfs(i,j);
if (lamb > wolf) res_lamb += lamb;
else res_wolf += wolf;
}
}
}
}
static void dfs(int x, int y){
visited[x][y] = true;
if (arr[x][y] == 'o') lamb++;
if (arr[x][y] == 'v') wolf++;
for (int i = 0; i < dir.length; i++){
int nx = x + dir[i][0];
int ny = y + dir[i][1];
if (nx < 0 || ny < 0 || nx >= N || ny >= M) continue;
if (visited[nx][ny] || arr[nx][ny] == '#') continue;
dfs(nx,ny);
}
}
public static void main(String[] args) throws IOException {
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
s = br.readLine();
st = new StringTokenizer(s);
N = Integer.parseInt(st.nextToken());
M = Integer.parseInt(st.nextToken());
arr = new char[N][M];
visited = new boolean[N][M];
for (int i = 0; i < N; i++){
s = br.readLine();
for (int j = 0; j < M; j++) arr[i][j] = s.charAt(j);
}
solution();
System.out.println(res_lamb + " " + res_wolf);
}
}
'자바 알고리즘 > 백준' 카테고리의 다른 글
백준 11725 : 트리의 부모 (0) | 2024.04.09 |
---|---|
백준 2606 : 바이러스 (0) | 2024.04.09 |
백준 4963 : 섬의 개수 (0) | 2024.04.09 |
백준 11724 : 연결 요소의 개수 (그래프 탐색) (0) | 2024.04.09 |
백준 1012 유기농 배추 (DFS) (0) | 2024.04.08 |