알고리듬/문제

[baekjoon] 14502

narmeee 2023. 3. 21. 17:10

연구소 

문제

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

연구소는 크기가 N×M인 직사각형으로 나타낼 수 있으며, 직사각형은 1×1 크기의 정사각형으로 나누어져 있다. 연구소는 빈 칸, 벽으로 이루어져 있으며, 벽은 칸 하나를 가득 차지한다. 

일부 칸은 바이러스가 존재하며, 이 바이러스는 상하좌우로 인접한 빈 칸으로 모두 퍼져나갈 수 있다. 새로 세울 수 있는 벽의 개수는 3개이며, 꼭 3개를 세워야 한다.

예를 들어, 아래와 같이 연구소가 생긴 경우를 살펴보자.

2 0 0 0 1 1 0
0 0 1 0 1 2 0
0 1 1 0 1 0 0
0 1 0 0 0 0 0
0 0 0 0 0 1 1
0 1 0 0 0 0 0
0 1 0 0 0 0 0

이때, 0은 빈 칸, 1은 벽, 2는 바이러스가 있는 곳이다. 아무런 벽을 세우지 않는다면, 바이러스는 모든 빈 칸으로 퍼져나갈 수 있다.

2행 1열, 1행 2열, 4행 6열에 벽을 세운다면 지도의 모양은 아래와 같아지게 된다.

2 1 0 0 1 1 0
1 0 1 0 1 2 0
0 1 1 0 1 0 0
0 1 0 0 0 1 0
0 0 0 0 0 1 1
0 1 0 0 0 0 0
0 1 0 0 0 0 0

바이러스가 퍼진 뒤의 모습은 아래와 같아진다.

2 1 0 0 1 1 2
1 0 1 0 1 2 2
0 1 1 0 1 2 2
0 1 0 0 0 1 2
0 0 0 0 0 1 1
0 1 0 0 0 0 0
0 1 0 0 0 0 0

벽을 3개 세운 뒤, 바이러스가 퍼질 수 없는 곳을 안전 영역이라고 한다. 위의 지도에서 안전 영역의 크기는 27이다.

연구소의 지도가 주어졌을 때 얻을 수 있는 안전 영역 크기의 최댓값을 구하는 프로그램을 작성하시오.

입력

첫째 줄에 지도의 세로 크기 N과 가로 크기 M이 주어진다. (3 ≤ N, M ≤ 8)

둘째 줄부터 N개의 줄에 지도의 모양이 주어진다. 0은 빈 칸, 1은 벽, 2는 바이러스가 있는 위치이다. 2의 개수는 2보다 크거나 같고, 10보다 작거나 같은 자연수이다.

빈 칸의 개수는 3개 이상이다.

출력

첫째 줄에 얻을 수 있는 안전 영역의 최대 크기를 출력한다.

예제 입력 1 

7 7
2 0 0 0 1 1 0
0 0 1 0 1 2 0
0 1 1 0 1 0 0
0 1 0 0 0 0 0
0 0 0 0 0 1 1
0 1 0 0 0 0 0
0 1 0 0 0 0 0

예제 출력 1 

27

풀이

빈 칸(0)들로 만들 수 있는 3개의 조합을 모두 탐색하면서 안전 영역의 크기를 구한다.

안전 영역의 크기를 구할 때는 시작점을 기준으로 연결되는 빈 칸(0)의 모든 개수를 구하고 바이러스(2)를 만나게되면 개수를 0으로 처리한다.

 

#include <iostream>

using namespace std;

#define LAND 0
#define WALL 1
#define VIRUS 2

struct node{
    int type;
    int visit;
};

int n, m;

void init(node ** board){
    for(int y = 0; y < n; y++){
        for(int x = 0; x < m; x++)board[y][x].visit = 0;
    }
}

//감염 : 0 , 감염 x : 1
int safeZone, infected;

void search(node ** board, int x, int y){
    if(y == n || y == -1 || x == m || x == -1)return;
    if(board[y][x].type == VIRUS){
        infected = 0;
        return;
    }
    if(board[y][x].type == WALL || board[y][x].visit == 1)return;

    board[y][x].visit = 1;
    safeZone++;

    search(board, x + 1, y);
    search(board, x, y + 1);
    search(board, x - 1, y);
    search(board, x, y - 1);
}

int countSafeZone(node ** board){
    int sum = 0;
    safeZone = 0;
    infected = 1;
    for(int y = 0; y < n; y++){
        for(int x = 0; x < m; x++){
            if(board[y][x].type != LAND || board[y][x].visit == 1)continue;
                search(board, x, y);
                sum += (safeZone * infected);
                safeZone = 0;
                infected = 1;
        }
    }
    return sum;
}

int main()
{
    
    node **board;
    cin >> n >> m;

    board = new node *[n];

    for(int i = 0; i < n; i++){
        board[i] = new node[m];
    }

    for(int y = 0; y < n; y++){
        for(int x = 0; x < m; x++){
            cin >> board[y][x].type;
            board[y][x].visit = 0;
        }
    }


    //벽 3개를 세울 수 있는 모든 경우 대입
    int max = 0, tmp;
    for(int p1 = 0; p1 < n * m - 2; p1++){
        //board[p1/m][p1%m]
        if(board[p1/m][p1%m].type != LAND)continue;
        board[p1/m][p1%m].type = WALL;

        for(int p2 = p1 + 1; p2 < n * m - 1; p2++){
            //board[p2/m][p2%m]
            if(board[p2/m][p2%m].type != LAND)continue;
            board[p2/m][p2%m].type = WALL;

            for(int p3 = p2 + 1; p3 < n *m; p3++){
                //board[p3/m][p3%m]
                if(board[p3/m][p3%m].type != LAND)continue;
                board[p3/m][p3%m].type = WALL;
                tmp = countSafeZone(board);
                init(board);
                if(tmp > max)max = tmp;

                board[p3/m][p3%m].type = LAND;
            }
            board[p2/m][p2%m].type = LAND;
        }
        board[p1/m][p1%m].type = LAND;
    }

    cout << max;
}

'알고리듬 > 문제' 카테고리의 다른 글

[baekjoon] 1194  (0) 2023.04.10
[baekjoon] 12100  (0) 2023.03.23
[baekjoon] 16234  (0) 2023.03.20
[baekjoon] 1090  (0) 2023.03.18
[baekjoon] 21315  (0) 2023.03.13