본문 바로가기
카테고리 없음

[알고리즘] c++ cpp 치즈

by keel_im 2020. 10. 12.
반응형

포인트

1. bfs 를 잘 할 수 있는가?

2. 아이디어는 중요한 것 같다. (외부 공기는 1개로 연결되어 있다. )

- 외부 공기 표시 ->(외부 공기 세기-> 치즈 녹이기 -> 외부 공기 다시 표시 (loop)) -> 결과

 

🧶문서는 항상 수정 될 수 있습니다. 비판은 환영합니다. 

#include<iostream>
#include<queue>
#include <tuple>

#define MAX 100
using namespace std;

int n, m;
int map[MAX][MAX];
int visited[MAX][MAX];

int dx[] = {0, 0, 1, -1};
int dy[] = {1, -1, 0, 0};
queue<pair<int, int>> nq;


void outAir() {
    queue<pair<int, int>> q; /* 외부 공기와 내부 공기를 분리하는 작업. */
    q.push(make_pair(0, 0)); /* visited[][] 배열의 초기값은 0 이고, 이 중에서 외부공기만 1로 바꿔준다. */
    visited[0][0] = 1;

    while (!q.empty()) {
        int x, y;
        tie(x, y) = q.front();
        q.pop();

        for (int i = 0; i < 4; i++) {
            int nx = x + dx[i];
            int ny = y + dy[i];

            if (nx < 0 || ny < 0 || nx >= n || ny >= m) continue; // 범위 체크

            if (map[nx][ny] == 0 && !visited[nx][ny]) {
                q.push({nx, ny});
                visited[nx][ny] = 1;
            }
        }
    }
}

void divideAir() {
    /* 외부 공기 중에서도, 치즈와 인접한 공기와 그렇지 않은 공기를 나누는 작업. */
    /* 치즈와 인접한 공기는 'nq'에 저장이 된다. */
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (visited[i][j] == 1) {
                for (int k = 0; k < 4; k++) {
                    int nx = i + dx[k];
                    int ny = j + dy[k];
                    if (nx < 0 || ny < 0 || nx >= n || ny >= m) continue;

                    if (map[nx][ny] == 1) {
                        nq.push(make_pair(i, j));
                        break;
                    }
                }
            }
        }
    }
}

void melt() {
    queue<pair<int, int>> q = nq;
    while (!nq.empty()) nq.pop();
    while (!q.empty()) {
        int x, y;
        tie(x, y) = q.front();
        q.pop();

        for (int i = 0; i < 4; i++) {
            int nx = x + dx[i];
            int ny = y + dy[i];
            if (nx < 0 || ny < 0 || nx >= n || ny >= m) continue;

            if (map[nx][ny] == 1) {
                map[nx][ny] = 0;
                nq.push(make_pair(nx, ny));
            }
        }
    }
}

void addAir() {
    // 내부 외부 // 인접 공기 nq 저장 --> bfs

    queue<pair<int, int>> q = nq; //복사
    while (!q.empty()) {
        int x, y;
        tie(x, y) = q.front();
        q.pop();

        for (int i = 0; i < 4; i++) {
            int nx = x + dx[i];
            int ny = y + dy[i];
            if (nx < 0 || ny < 0 || nx >= n || ny >= m) continue; // 범위 체크

            if (!visited[nx][ny]) {
                q.push(make_pair(nx, ny));
                nq.push(make_pair(nx, ny));
                visited[nx][ny] = 1;
            }
        }
    }
}

bool check() {
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (map[i][j] == 1) return false; // 치즈가 1개라도 있으면 false
        }
    }
    return true;
}

int main() {
    cin >> n >> m;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            cin >> map[i][j];
            if (map[i][j] == 1) {
                visited[i][j] = -1;    // 입력과 동시에 치즈가 있는 좌표는 -1로 표시
            }
        }
    }
    outAir();
    divideAir();

    int time = 0;
    int size = 0;

    while (1) {
        if (check()) break;
        melt();
        size = nq.size();

        addAir();
        time+=1;
    }
    cout << time << '\n' << size << '\n';
    return 0;
}

 

반응형

댓글