본문 바로가기
Algorithm/구현

[백준] 14502번 : 연구소 (삼성 SW 역량 테스트)

by 루시킴 2021. 11. 2.

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

 

14502번: 연구소

인체에 치명적인 바이러스를 연구하던 연구소에서 바이러스가 유출되었다. 다행히 바이러스는 아직 퍼지지 않았고, 바이러스의 확산을 막기 위해서 연구소에 벽을 세우려고 한다. 연구소는 크

www.acmicpc.net

#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>
using namespace std;

int map[8][8];
int map_cp[8][8];
int dx[4] = { -1,1,0,0 };
int dy[4] = { 0,0,-1,1 };
int N, M;
int ans=0;
vector <pair<int, int>> virus;

void bfs() {
	queue<pair<int, int>> q;
	for (int i = 0; i < virus.size(); i++){
		q.push(make_pair(virus[i].first, virus[i].second));
	}


	while (!q.empty()) {
		int x = q.front().first;
		int y = q.front().second;
		q.pop();

		for (int i = 0; i < 4; i++){
			int cx = x + dx[i];
			int cy = y + dy[i];
			if (cx >= 0 && cx <= N-1 && cy >= 0 && cy <= M-1) {
				if (!map_cp[cx][cy]) {
					map_cp[cx][cy] = 2;
					q.push(make_pair(cx, cy));
					
				}
			}
		}
	}
	int tmp = 0;
	for (int i = 0; i < N; i++){
		for (int j = 0; j < M; j++) {
			if (map_cp[i][j] == 0) {
				tmp++;
			}
		}
	}
	ans = max(ans, tmp);
	
}


void makewall(int cnt) {
	if (cnt == 3) {
		for (int i = 0; i < N; i++){
			for (int j = 0; j < M; j++) {
				map_cp[i][j] = map[i][j];
			}
		}
		bfs();
		return;
	}

	for (int i = 0; i < N; i++){
		for (int j = 0; j < M; j++) {
			if (map[i][j] == 0) {
				map[i][j] = 1;
				makewall(cnt + 1);
				map[i][j] = 0;
			}
		}
	}
}

int main() {
	cin >> N >> M;
	for (int i = 0; i < N; i++){
		for (int j = 0; j < M; j++) {
			cin >> map[i][j];
			if (map[i][j] == 2) virus.push_back(make_pair(i, j));

		}
	}

	for (int i = 0; i < N; i++){
		for (int j = 0; j < M; j++) {
			if (map[i][j] == 0) {
				map[i][j] = 1;
				makewall(1);
				map[i][j] = 0;
			}
		}
	}

	cout << ans;

	return 0;
}

댓글