알고리즘/PS - 백준

[백준 2638 - C++] 치즈 : BFS

excited-hyun 2021. 4. 11. 21:04
반응형

www.acmicpc.net/problem/2638

 

2638번: 치즈

첫째 줄에는 모눈종이의 크기를 나타내는 두 개의 정수 N, M (5≤N, M≤100)이 주어진다. 그 다음 N개의 줄에는 모눈종이 위의 격자에 치즈가 있는 부분은 1로 표시되고, 치즈가 없는 부분은 0으로 표

www.acmicpc.net

문제

N×M (5≤N, M≤100)의 모눈종이 위에 아주 얇은 치즈가 <그림 1>과 같이 표시되어 있다. 단, N 은 세로 격자의 수이고, M 은 가로 격자의 수이다. 이 치즈는 냉동 보관을 해야만 하는데 실내온도에 내어놓으면 공기와 접촉하여 천천히 녹는다. 그런데 이러한 모눈종이 모양의 치즈에서 각 치즈 격자(작 은 정사각형 모양)의 4변 중에서 적어도 2변 이상이 실내온도의 공기와 접촉한 것은 정확히 한시간만에 녹아 없어져 버린다. 따라서 아래 <그림 1> 모양과 같은 치즈(회색으로 표시된 부분)라면 C로 표시된 모든 치즈 격자는 한 시간 후에 사라진다.

<그림 2>와 같이 치즈 내부에 있는 공간은 치즈 외부 공기와 접촉하지 않는 것으로 가정한다. 그러므 로 이 공간에 접촉한 치즈 격자는 녹지 않고 C로 표시된 치즈 격자만 사라진다. 그러나 한 시간 후, 이 공간으로 외부공기가 유입되면 <그림 3>에서와 같이 C로 표시된 치즈 격자들이 사라지게 된다.

모눈종이의 맨 가장자리에는 치즈가 놓이지 않는 것으로 가정한다. 입력으로 주어진 치즈가 모두 녹아 없어지는데 걸리는 정확한 시간을 구하는 프로그램을 작성하시오.

 

입력

첫째 줄에는 모눈종이의 크기를 나타내는 두 개의 정수 N, M (5≤N, M≤100)이 주어진다. 그 다음 N개의 줄에는 모눈종이 위의 격자에 치즈가 있는 부분은 1로 표시되고, 치즈가 없는 부분은 0으로 표시된다. 또한, 각 0과 1은 하나의 공백으로 분리되어 있다.

 

출력

출력으로는 주어진 치즈가 모두 녹아 없어지는데 걸리는 정확한 시간을 정수로 첫 줄에 출력한다.

 

 

풀이

이 문제에서는 외부공간에 대해서 bfs를 수행하면서 외부 공간으로 연결된 모든 위치의 map[][]값은 -1로 넣어주고 수행해야한다.

그래서 이 역할을 하는 함수를 따로 만들었다.

하루가 지날 때마다 두변 이상이 공기 접촉된 치즈는 사라진다. 그렇기 때믄에 외부공간의 연결을 확인하는 함수를 하루가 시작할때 매번 호출해야한다.

코드상으로는 치즈가 왼쪽 위부터 사라지게 구현하였지만 사실은 모두 같은 하루에 사라져야한다. 만약 여기서 그냥 치즈가 사라진 위치에 -1을 넣는 다면 왼쪽위에 먼저 사라진 치즈의 위치까지 치즈가 없는 외부 공간으로 인식하여 그 오른쪽, 아래줄 들에서 문제가 생길 수 있다.

따라서 0을 넣어준뒤, 치즈의 삭제여부를 모두 체크하고 난 뒤 외부 공간의 연결을 확인하는 함수를 또 호출해 주어야 한다.

 

#include <iostream>
#include <cstdio>
#include <queue>

using namespace std;

int X[4] = {1, -1, 0, 0};
int Y[4] = {0, 0, 1, -1};
queue<pair<int, int>> check;
int map[101][101];  //외부 = -1, 내부 = 0, 치즈 = 1
int visited[101][101];

void check_out(int n, int m);   //외부공간 확인하는 함수

int main (void){
    int n, m;
    int near_x, near_y;
    int cheese_cnt= 0;
    int day = 0;
    int cnt = 0;

    
    scanf("%d %d", &n, &m);
    
    for(int i=0; i<n; i++){
        for(int j=0; j<m; j++){
            scanf("%d", &map[i][j]);
            if(map[i][j] == 1)
                cheese_cnt++;
        }
    }

    while(cheese_cnt !=0){
        //printf("%d\n", cheese_cnt);
        day++;
        check_out(n, m);

        for(int i=0; i<n; i++){
            for(int j=0; j<m; j++){
                
                if(map[i][j] != 1)
                    continue;
                
                cnt = 0;
                for(int k=0; k<4; k++){
                    near_x = i + X[k];
                    near_y = j + Y[k];
                    if(near_x < 0 || near_x >= n || near_y <0 || near_y >=m)
                        continue;
                    if(map[near_x][near_y] == -1)
                        cnt++;
                    
                }
                //printf("%d\n", cnt);
                if(cnt >=2){
                    map[i][j] = 0;
                    cheese_cnt--;
                }
                
                if(cheese_cnt == 0){
                    printf("%d\n", day);
                    return 0;
                }
            }
        }
    }
}

void check_out(int n, int m){

    int found = 0;
    int s_x, s_y;
    int n_x, n_y;
    
    for(int i=0; i<n; i++){
        for(int j=0; j<m; j++){
            visited[i][j] = 0;
        }
    }
    
    //각각의 모서리 확인
    for(int i=0; i<n; i++){
        if(map[i][0] == 0 || map[i][0] == -1){
            found = 1;
            s_x = i;
            s_y = 0;
            break;
        }
    }
    if(found == 0){
        for(int j=0; j<m; j++){
            if(map[0][j] == 0 || map[0][j] == -1){
                found = 1;
                s_x = 0;
                s_y = j;
                break;
            }
        }
    }
    if(found == 0){
        for(int j=0; j<m; j++){
            if(map[n-1][j] == 0 || map[n-1][j] == -1){
                found = 1;
                s_x = n-1;
                s_y = j;
                break;
            }
        }
    }
    if(found == 0){
        for(int i=0; i<n; i++){
            if(map[i][m-1] == 0 || map[i][m-1] == -1){
                found = 1;
                s_x = i;
                s_y = m-1;
                break;
            }
        }
    }
    check.push(make_pair(s_x, s_y));
    visited[s_x][s_y] = 1;
    map[s_x][s_y] = -1;
    
    while(!check.empty()){
        s_x = check.front().first;
        s_y = check.front().second;
        check.pop();
        for(int i=0; i<4; i++){
            n_x = s_x + X[i];
            n_y = s_y + Y[i];
            if(n_x < 0 || n_x >= n || n_y <0 || n_y >=m)
                continue;
            if(visited[n_x][n_y] || map[n_x][n_y] == 1)
                continue;
            check.push(make_pair(n_x, n_y));
            visited[n_x][n_y] = 1;
            map[n_x][n_y] = -1;
        }
    }
    
}
728x90
반응형