👀 문제
https://www.acmicpc.net/problem/14502
👊 도전
1. 설계
- 조합을 이용해 벽 3개를 세운다.
- DFS로 바이러스를 퍼트린다.
- 안전구역은 max를 저장한다.
2. 구현 (성공 코드)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
import java.util.ArrayList;
import java.util.Scanner;
/**
* @author HEESOO
*
*/
class Node{
int x;
int y;
public Node(int x, int y) {
this.x=x;
this.y=y;
}
}
public class Main {
static int m, n, answer=0;
static int[][] map, temp;
static ArrayList<Node> virus;
public static void comb(int start, int depth) {
if(depth==3) {
copyMap();
for(int i=0;i<virus.size();i++) {
Node node=virus.get(i);
dfs(node.x, node.y);
}
zeroCount();
return;
}
else {
for(int i=start;i<n*m;i++) {
int x=i/m;
int y=i%m;
if(map[x][y]==0) {
map[x][y]=1;
comb(i+1, depth+1);
map[x][y]=0;
}
}
}
}
public static void copyMap() {
temp=new int[n][m];
for(int i=0;i<n;i++) {
for(int j=0;j<m;j++) {
temp[i][j]=map[i][j];
}
}
}
public static void dfs(int x, int y) {
int[] dotX= {0,0,-1,1};
int[] dotY= {1,-1,0,0};
for(int i=0;i<4;i++) {
int xx=x+dotX[i];
int yy=y+dotY[i];
if(0<=xx&&xx<n&&0<=yy&&yy<m) {
if(temp[xx][yy]==0) {
temp[xx][yy]=2;
dfs(xx,yy);
}
}
}
}
public static void zeroCount() {
int zero=0;
for(int i=0;i<n;i++) {
for(int j=0;j<m;j++) {
if(temp[i][j]==0) zero++;
}
}
answer=Math.max(answer, zero);
}
public static void main(String[] args) {
// TODO Auto-generated method stub
Scanner scan = new Scanner(System.in);
n=scan.nextInt();
m=scan.nextInt();
map=new int[n][m];
virus=new ArrayList<>();
for(int i=0;i<n;i++) {
for(int j=0;j<m;j++) {
map[i][j]=scan.nextInt();
if(map[i][j]==2) virus.add(new Node(i, j));
}
}
comb(0,0);
System.out.println(answer);
}
}
3. 결과
🤟 성공 🤟
4. 설명
- 조합을 이용해 벽 3개를 세운다
- 벽은 순서에 상관없이 3개만 뽑으면 되므로 조합을 이용한다.
- 2차원 배열이므로 n*m개의 인덱스를 순회해야하며, i/m, i%m으로 x, y를 계산한다.
- map[i][j]==0이면 1로 바꿔 벽을 세우고, 재귀호출하여 다음 벽을 찾는다. map[i][j]에 벽을 세우지 않을 수도 있으므로 재귀가 끝나고 난 다음에 0으로 다시 초기화하여 다음 사용을 가능하게 한다.
- depth==3이면 벽 3개를 세운 것과 같다.
- DFS로 바이러스를 퍼트린다
- 완성된 map은 다음 벽을 세우기 위해 계속 사용해야하므로 temp에 카피하여, temp를 가지고 바이러스를 퍼트린다.
- main에서 바이러스의 위치를 Node를 이용해 ArrayList에 저장해둔다.
- 바이러스들이 담긴 ArrayList를 순회하며 각 위치에서 상하좌우로 빈 칸에 바이러스를 퍼트린 다음, 그 위치에서 다시 dfs를 호출하여 바이러스가 퍼지도록 한다.
- 안전구역을 센다
- 바이러스가 다 퍼졌다면, zeroCount()를 호출하여 현재 temp의 빈 칸 갯수를 센다.
- answer에는 지금까지의 안전구역 갯수 중 가장 큰 값을 저장한다.
👏 해결 완료!
참고
- 백준 14502번. 연구소 (Java, Python) https://bcp0109.tistory.com/25