BOJ 15683 - 감시 [C++]

문제

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

 

스타트링크의 사무실은 1×1크기의 정사각형으로 나누어져 있는 N×M 크기의 직사각형으로 나타낼 수 있다. 사무실에는 총 K개의 CCTV가 설치되어져 있는데, CCTV는 5가지 종류가 있다. 각 CCTV가 감시할 수 있는 방법은 다음과 같다.

 

1번 CCTV는 한 쪽 방향만 감시할 수 있다. 2번과 3번은 두 방향을 감시할 수 있는데, 2번은 감시하는 방향이 서로 반대방향이어야 하고, 3번은 직각 방향이어야 한다. 4번은 세 방향, 5번은 네 방향을 감시할 수 있다.

CCTV는 감시할 수 있는 방향에 있는 칸 전체를 감시할 수 있다. 사무실에는 벽이 있는데, CCTV는 벽을 통과할 수 없다. CCTV가 감시할 수 없는 영역은 사각지대라고 한다.

CCTV는 회전시킬 수 있는데, 회전은 항상 90도 방향으로 해야 하며, 감시하려고 하는 방향이 가로 또는 세로 방향이어야 한다.

사무실의 크기와 상태, 그리고 CCTV의 정보가 주어졌을 때, CCTV의 방향을 적절히 정해서, 사각 지대의 최소 크기를 구하는 프로그램을 작성하시오.

 

풀이

CCTV의 최대 개수가 8개이고, 각 CCTV의 서로 다른 방향의 개수는 최대 4개이므로 모든 경우의 수에 대해 시뮬레이션을 하면 대략 $O(4^8 \times NM)$의 시간이 걸린다. 즉, 완전 탐색이 가능하다.

재귀로 CCTV들의 상태로 가능한 모든 경우를 구해줬고, 각각의 경우에 대해 감시 범위를 직접 체크해놓고 사각지대를 찾아주었다.

 

#include <bits/stdc++.h>
using namespace std;

int arr[8][8]{}, N, M, cc = 0, ans = 1e9;
vector<tuple<int, int, int, int>> CCTV;
int dx[4] = { 0,-1,0,1 };
int dy[4] = { 1,0,-1,0 };
int temp[8][8]{};

void move(int x, int y, int dir) {
	for (x += dx[dir], y += dy[dir]; x >= 0 && x < N && y >= 0 && y < M && temp[x][y] != 6; x += dx[dir], y += dy[dir]) {
		if (!temp[x][y])	temp[x][y] = -1;
	}
}

int cnt = 0;
void sol(int pos, vector<tuple<int, int, int, int>> state) {
	if (pos == CCTV.size()) {
		for (int i = 0; i < N; i++)	for (int j = 0; j < M; j++)	temp[i][j] = arr[i][j];
		for (auto [i, x, y, d] : state) {
			if (i == 1) {
				move(x, y, d);
			}
			else if (i == 2) {
				move(x, y, d);
				move(x, y, (d + 2) % 4);
			}
			else if (i == 3) {
				move(x, y, d);
				move(x, y, (d + 1) % 4);
			}
			else if (i == 4) {
				move(x, y, d);
				move(x, y, (d + 1) % 4);
				move(x, y, (d + 2) % 4);
			}
			else {
				for (int k = 0; k < 4; k++)	move(x, y, k);
			}
		}
		int res = 0;
		for (int i = 0; i < N; i++) {
			for (int j = 0; j < M; j++) {
				if (!temp[i][j])	res++;
			}
		}
		ans = min(ans, res);
		return;
	}
	auto [a, b, c, d] = CCTV[pos];
	state.emplace_back(a, b, c, 0);
	sol(pos + 1, state);
	state.pop_back();
	state.emplace_back(a, b, c, 1);
	sol(pos + 1, state);
	state.pop_back();
	state.emplace_back(a, b, c, 2);
	sol(pos + 1, state);
	state.pop_back();
	state.emplace_back(a, b, c, 3);
	sol(pos + 1, state);
	state.pop_back();
}

int main() {
	cin.tie(0)->sync_with_stdio(0);
	
	cin >> N >> M;
	for (int i = 0; i < N; i++)	for (int j = 0; j < M; j++)	cin >> arr[i][j];

	for (int i = 0; i < N; i++) {
		for (int j = 0; j < M; j++) {
			if (1 <= arr[i][j] && arr[i][j] <= 5)	CCTV.emplace_back(arr[i][j], i, j, 0);
		}
	}

	sol(0, {});
	cout << ans;

}