[JAVA/백준] 삼성 SW 역량 테스트 기출 문제: 연구소

👀 문제

https://www.acmicpc.net/problem/14502

👊 도전

1. 설계

  1. 조합을 이용해 벽 3개를 세운다.
  2. DFS로 바이러스를 퍼트린다.
  3. 안전구역은 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. 설명

  1. 조합을 이용해 벽 3개를 세운다
    • 벽은 순서에 상관없이 3개만 뽑으면 되므로 조합을 이용한다.
    • 2차원 배열이므로 n*m개의 인덱스를 순회해야하며, i/m, i%m으로 x, y를 계산한다.
    • map[i][j]==0이면 1로 바꿔 벽을 세우고, 재귀호출하여 다음 벽을 찾는다. map[i][j]에 벽을 세우지 않을 수도 있으므로 재귀가 끝나고 난 다음에 0으로 다시 초기화하여 다음 사용을 가능하게 한다.
    • depth==3이면 벽 3개를 세운 것과 같다.
  2. DFS로 바이러스를 퍼트린다
    • 완성된 map은 다음 벽을 세우기 위해 계속 사용해야하므로 temp에 카피하여, temp를 가지고 바이러스를 퍼트린다.
    • main에서 바이러스의 위치를 Node를 이용해 ArrayList에 저장해둔다.
    • 바이러스들이 담긴 ArrayList를 순회하며 각 위치에서 상하좌우로 빈 칸에 바이러스를 퍼트린 다음, 그 위치에서 다시 dfs를 호출하여 바이러스가 퍼지도록 한다.
  3. 안전구역을 센다
    • 바이러스가 다 퍼졌다면, zeroCount()를 호출하여 현재 temp의 빈 칸 갯수를 센다.
    • answer에는 지금까지의 안전구역 갯수 중 가장 큰 값을 저장한다.

👏 해결 완료!

참고