[백준] 14502 연구소

0

백준

목록 보기
225/271
post-thumbnail

[백준] 14502 연구소

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

int N,M;
int board[101][101] = { 0 };

//바이러스 좌표
vector<pair<int, int>> virus;

//최대 안전영역 크기
int maxSafe = -1;


bool inRange(int r, int c) {
	if (r < 0 || r >= N) return false;
	if (c < 0 || c >= M) return false;
	return true;
}

int dirR[4] = { 0, 0, 1, -1 };
int dirC[4] = { 1, -1, 0, 0 };

//안전한 영역의 크기 반환
int getSafe() {
	int visited[101][101] = { 0 };
	queue<pair<int, int>> q;
	
	//bfs 시작 좌표 = 바이러스의 좌표
	for (int i = 0; i < virus.size(); ++i) {
		q.push(virus[i]);
	}

	while (!q.empty()) {
		int curR = q.front().first; 
		int curC = q.front().second;
		q.pop();

		if (visited[curR][curC]) continue;
		visited[curR][curC] = 1;

		//상하좌우 인접한 칸 bfs
		for (int d = 0; d < 4; ++d) {
			int nextR = curR + dirR[d];
			int nextC = curC + dirC[d];

			if (!inRange(nextR, nextC)) continue;
			if (visited[nextR][nextC]) continue;
			if (board[nextR][nextC] == 1) continue;

			q.push({ nextR, nextC });
		}
	}
    
	//벽이 아니면서 바이러스가 퍼지지 않은 좌표의 수 = 안전 영역의 크기
	int safe = 0;
	for (int i = 0; i < N; ++i) {
		for (int j = 0; j < M; ++j) {
			if ((board[i][j] == 0) && (visited[i][j] == 0)) 
				safe++;
		}
	}

	return safe;
}


void chooseWall(int wallCnt) {
	if (wallCnt == 3) {
		maxSafe = max(maxSafe, getSafe());
		return;
	}

	//벽 고르기
	for (int r = 0; r < N; ++r) {
		for (int c = 0; c < M; ++c) {
			if (board[r][c] == 0){
				board[r][c] = 1;
				chooseWall(wallCnt + 1);
				board[r][c] = 0;
			}
		}
	}

	return;
}

int main() {
	ios_base::sync_with_stdio(false);
	cin.tie(NULL); cout.tie(NULL);
	
	cin >> N >> M;

	for (int i = 0; i < N; ++i) {
		for (int j = 0; j < M; ++j) {
			int input;
			cin >> input;
			board[i][j] = input;

			if (board[i][j] == 2) {
				virus.push_back({i, j});
			}
		}
	}

	chooseWall(0);

	cout << maxSafe;
	return 0;
}

profile
Be able to be vulnerable, in search of truth

0개의 댓글