연구소 3

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

17142번: 연구소 3

 

조합을 이용해서 활성 바이러스를 고르고
최소 시간이니까 BFS를 하자.
jikan을 visited로 쓰면 되겠다.

초기에 입력을 받을 때 빈 공간의 갯수를 Count해주었다.

탐색을 진행하면서, 빈 공간을 탐색할 때 마다 그 개수를 Count해줄 것이고,

이 개수와 처음에 Count해놓은 빈 공간의 개수와 비교해서 모두 탐색이 가능한지 확인

 

추가 조건

활성 바이러스가 비활성 바이러스가 있는 칸으로 가면

비활성 바이러스가 활성으로 변한다.

비활성 바이러스를 만나면 시간 추가를 안한다....

#include <iostream>
#include <vector>
#include <queue>
#include <cstring>
 
using namespace std;
 
int dx[] = { 0, 0, -1, 1 };
int dy[] = { 1, -1, 0, 0 };
 
int n, m;
int aki;
int ans = 987654321;
 
vector<pair<int, int>> virus;
int map[50][50];
//visited의 기능까지 한다!
int jikan[50][50];
bool chk[15];
 
void bfs(queue<pair<int,int>> q)
{
    int infect = 0;
    // 0을 갈 때만 total 갱신
    int total = 0;
    while (!q.empty())
    {
        int x = q.front().first;
        int y = q.front().second;
        q.pop();
 
        for (int k = 0; k < 4; k++)
        {
            int nx = x + dx[k];
            int ny = y + dy[k];
 
            if (nx >= 0 && nx < n && ny >= 0 && ny < n)
            {
                if (map[nx][ny] != 1 && jikan[nx][ny] == -1)
                {
                    jikan[nx][ny] = jikan[x][y] + 1;
                    if (map[nx][ny] == 0)
                    {
                        infect++;
                        total = jikan[nx][ny];
                    }
                    q.push(make_pair(nx, ny));
                }
            }
        }
    }
    if (infect == aki)
    {
        if (ans > total)
            ans = total;
    }
 
}
 
void go(int index, int cnt)
{
    if (cnt == m)
    {
        queue<pair<int, int>> q;
        memset(jikan, -1, sizeof(jikan));
        for (int i = 0; i < virus.size(); i++)
        {
            if (chk[i])
            {
                q.push(make_pair(virus[i].first, virus[i].second));
                jikan[virus[i].first][virus[i].second] = 0;
            }
        }
 
        bfs(q);
        return;
    }
 
    for (int i = index; i < virus.size(); i++)
    {
        if (chk[i])
            continue;
        chk[i] = true;
        go(i + 1, cnt + 1);
        chk[i] = false;
    }
}
 
int main()
{
    ios::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 < n; j++)
        {
            cin >> map[i][j];
            if (map[i][j] == 0)
                aki++;
            else if (map[i][j] == 2)
            {
                virus.push_back({ i, j });
            }
        }
    }
 
    go(0, 0);
 
    if (ans == 987654321)
        cout << -1 << '\n';
    else
        cout << ans << '\n';
 
    return 0;
}