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

[알고리즘] c++ cpp 빙산

by keel_im 2020. 10. 12.
반응형

포인트

1. bfs를 잘 이해할 수 있는가?

2. map을 복사해서 잘 사용을 할 수 있는가?

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

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

using namespace std;

int n, m;
int map[300][300];
int cmap[300][300];
bool visited[300][300];

int dx[] = {0, 0, 1, -1};
int dy[] = {1, -1, 0, 0};

void bfs(int a, int b) {
    queue<pair<int, int>> q;
    q.push({a, b});
    visited[a][b] = 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 (!visited[nx][ny] && map[nx][ny] != 0) {
                visited[nx][ny] = 1;
                q.push(make_pair(nx, ny));
            }
        }
    }
}

int melt(int x, int y) { // 4방향 물인 것들 세기
    int cnt = 0;
    for (int i = 0; i < 4; i++) {
        int nx = x + dx[i];
        int ny = y + dy[i];

        if (map[nx][ny] == 0) cnt+=1;
    }
    return cnt;
}

int main() {
    cin >> n >> m;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            cin >> map[i][j];
        }
    }

    int year = 0;
    while (1) {
        int cnt = 0;
        memset(visited, 0, sizeof(visited));
        memset(cmap, 0, sizeof(cmap));

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (map[i][j] != 0 && visited[i][j] == 0) {
                    cnt += 1;
                    bfs(i, j);
                }
            }
        }

        if (cnt >= 2) {
            cout << year << '\n';
            return 0;
        }
        if (cnt == 0) {
            cout << 0 << '\n';
            return 0;
        }

        for (int i = 0; i < n; i++) {
            for (int j = 0; j < m; j++) {
                if (map[i][j] != 0) {
                    cmap[i][j] = map[i][j] - melt(i, j);
                    if (cmap[i][j] < 0) cmap[i][j] = 0;
                }
            }
        }

        memcpy(map, cmap, sizeof(map)); //녹은거 반영
        year+=1;
    }

}

 

반응형

댓글