답안 #996991

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
996991 2024-06-11T14:42:09 Z imann Mecho (IOI09_mecho) C++17
50 / 100
132 ms 6492 KB
#include <iostream>
#include <array>
#include <string>
#include <queue>

const int MAX_N = 800 + 1;
const int INF = 1e9;

std::array<std::string, MAX_N> grid;
std::array<std::array<int, MAX_N>, MAX_N> minuteToBeHive, level;

void bfs(int n) {
    std::queue<std::pair<int, int>> q;
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            minuteToBeHive[i][j] = INF;
            if (grid[i][j] == 'H') {
                q.push({i, j});
                minuteToBeHive[i][j] = 0;
            }
        }
    }

    while (!q.empty()) {
        auto v = q.front();
        q.pop();
        if (v.first > 0) {
            if (grid[v.first - 1][v.second] == 'G' || grid[v.first - 1][v.second] == 'M') {
                if (minuteToBeHive[v.first - 1][v.second] == INF)
                    q.push({v.first - 1, v.second});
                minuteToBeHive[v.first - 1][v.second] = std::min(minuteToBeHive[v.first - 1][v.second], minuteToBeHive[v.first][v.second] + 1);
            }
        }
        if (v.first < n - 1) {
            if (grid[v.first + 1][v.second] == 'G' || grid[v.first + 1][v.second] == 'M') {
                if (minuteToBeHive[v.first + 1][v.second] == INF)
                    q.push({v.first + 1, v.second});
                minuteToBeHive[v.first + 1][v.second] = std::min(minuteToBeHive[v.first + 1][v.second], minuteToBeHive[v.first][v.second] + 1);
            }
        }
        if (v.second > 0) {
            if (grid[v.first][v.second - 1] == 'G' || grid[v.first][v.second - 1] == 'M') {
                if (minuteToBeHive[v.first][v.second - 1] == INF)
                    q.push({v.first, v.second - 1});
                minuteToBeHive[v.first][v.second - 1] = std::min(minuteToBeHive[v.first][v.second - 1], minuteToBeHive[v.first][v.second] + 1);
            }
        }
        if (v.second < n - 1) {
            if (grid[v.first][v.second + 1] == 'G' || grid[v.first][v.second + 1] == 'M') {
                if (minuteToBeHive[v.first][v.second + 1] == INF)
                    q.push({v.first, v.second + 1});
                minuteToBeHive[v.first][v.second + 1] = std::min(minuteToBeHive[v.first][v.second + 1], minuteToBeHive[v.first][v.second] + 1);
            }
        }
    }
}

bool bfs2(int n, int s, int startMinute) {
    std::pair<int, int> start, end;
    std::queue<std::pair<int, int>> q;
    for (int i = 0; i < n; ++i) {
        for (int j = 0; j < n; ++j) {
            level[i][j] = INF;
            if (grid[i][j] == 'M')
                start = {i, j};
            if (grid[i][j] == 'D')
                end = {i, j};
        }
    }
    q.push(start);
    level[start.first][start.second] = 0;
    if (startMinute >= minuteToBeHive[start.first][start.second])
        return false;
    while (!q.empty()) {
        auto v = q.front();
        q.pop();
        if (v.first > 0) {
            if ((grid[v.first - 1][v.second] == 'G' || grid[v.first - 1][v.second] == 'D' || grid[v.first - 1][v.second] == 'M') && (startMinute + (level[v.first][v.second] + 1) / s < minuteToBeHive[v.first - 1][v.second])) {
                if (level[v.first - 1][v.second] == INF)
                    q.push({v.first - 1, v.second});
                level[v.first - 1][v.second] = std::min(level[v.first - 1][v.second], level[v.first][v.second] + 1);
            }
        }
        if (v.first < n - 1) {
            if ((grid[v.first + 1][v.second] == 'G' || grid[v.first + 1][v.second] == 'D' || grid[v.first + 1][v.second] == 'M') && (startMinute + (level[v.first][v.second] + 1) / s < minuteToBeHive[v.first + 1][v.second])) {
                if (level[v.first + 1][v.second] == INF)
                    q.push({v.first + 1, v.second});
                level[v.first + 1][v.second] = std::min(level[v.first + 1][v.second], level[v.first][v.second] + 1);
            }
        }
        if (v.second > 0) {
            if ((grid[v.first][v.second - 1] == 'G' || grid[v.first][v.second - 1] == 'D' || grid[v.first][v.second - 1] == 'M') && (startMinute + (level[v.first][v.second] + 1) / s < minuteToBeHive[v.first][v.second - 1])) {
                if (level[v.first][v.second - 1] == INF)
                    q.push({v.first, v.second - 1});
                level[v.first][v.second - 1] = std::min(level[v.first][v.second - 1], level[v.first][v.second] + 1);
            }
        }
        if (v.second < n - 1) {
            if ((grid[v.first][v.second + 1] == 'G' || grid[v.first][v.second + 1] == 'D' || grid[v.first][v.second + 1] == 'M') && (startMinute + (level[v.first][v.second] + 1) / s < minuteToBeHive[v.first][v.second + 1])) {
                if (level[v.first][v.second + 1] == INF)
                    q.push({v.first, v.second + 1});
                level[v.first][v.second + 1] = std::min(level[v.first][v.second + 1], level[v.first][v.second] + 1);
            }
        }
    }
    return level[end.first][end.second] != INF;
}

int solve(int n, int s) {
    bfs(n);
    int l = 0, r = n;
    while (r - l > 1) {
        int mid = (r + l) / 2;
        if (bfs2(n, s, mid)) {
            l = mid;
        } else {
            r = mid;
        }
    }

    if (bfs2(n, s, l))
        return l;
    return -1;
}

int main() {
    int n, s; std::cin >> n >> s;
    for (int i = 0; i < n; ++i) {
        std::cin >> grid[i];
    }
    std::cout << solve(n, s) << std::endl;
}
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 2396 KB Output is correct
2 Correct 0 ms 2396 KB Output is correct
3 Correct 1 ms 2396 KB Output is correct
4 Correct 1 ms 2396 KB Output is correct
5 Correct 0 ms 2396 KB Output is correct
6 Correct 0 ms 2396 KB Output is correct
7 Correct 61 ms 6488 KB Output is correct
8 Correct 0 ms 2396 KB Output is correct
9 Correct 1 ms 2648 KB Output is correct
10 Correct 0 ms 2392 KB Output is correct
11 Correct 1 ms 2396 KB Output is correct
12 Incorrect 1 ms 4700 KB Output isn't correct
13 Correct 1 ms 2652 KB Output is correct
14 Correct 1 ms 4700 KB Output is correct
15 Incorrect 0 ms 2396 KB Output isn't correct
16 Incorrect 1 ms 2396 KB Output isn't correct
17 Incorrect 1 ms 2396 KB Output isn't correct
18 Incorrect 1 ms 2396 KB Output isn't correct
19 Incorrect 1 ms 2396 KB Output isn't correct
20 Incorrect 1 ms 2396 KB Output isn't correct
21 Incorrect 1 ms 2520 KB Output isn't correct
22 Incorrect 0 ms 2396 KB Output isn't correct
23 Incorrect 1 ms 2652 KB Output isn't correct
24 Incorrect 1 ms 2652 KB Output isn't correct
25 Incorrect 1 ms 4700 KB Output isn't correct
26 Incorrect 1 ms 4700 KB Output isn't correct
27 Incorrect 1 ms 4700 KB Output isn't correct
28 Incorrect 1 ms 4756 KB Output isn't correct
29 Incorrect 1 ms 4700 KB Output isn't correct
30 Incorrect 1 ms 4700 KB Output isn't correct
31 Incorrect 1 ms 4700 KB Output isn't correct
32 Incorrect 1 ms 4700 KB Output isn't correct
33 Incorrect 10 ms 4696 KB Output isn't correct
34 Incorrect 12 ms 4912 KB Output isn't correct
35 Correct 16 ms 4700 KB Output is correct
36 Incorrect 12 ms 4944 KB Output isn't correct
37 Incorrect 14 ms 4696 KB Output isn't correct
38 Correct 23 ms 4696 KB Output is correct
39 Incorrect 15 ms 4956 KB Output isn't correct
40 Incorrect 18 ms 4964 KB Output isn't correct
41 Correct 29 ms 4944 KB Output is correct
42 Incorrect 21 ms 5228 KB Output isn't correct
43 Incorrect 24 ms 5212 KB Output isn't correct
44 Correct 32 ms 5212 KB Output is correct
45 Incorrect 31 ms 5244 KB Output isn't correct
46 Incorrect 28 ms 5212 KB Output isn't correct
47 Correct 38 ms 5208 KB Output is correct
48 Incorrect 29 ms 5428 KB Output isn't correct
49 Incorrect 33 ms 5268 KB Output isn't correct
50 Correct 51 ms 5212 KB Output is correct
51 Incorrect 34 ms 5464 KB Output isn't correct
52 Incorrect 40 ms 5468 KB Output isn't correct
53 Correct 60 ms 5604 KB Output is correct
54 Incorrect 39 ms 5720 KB Output isn't correct
55 Incorrect 46 ms 5724 KB Output isn't correct
56 Correct 68 ms 5820 KB Output is correct
57 Incorrect 44 ms 5980 KB Output isn't correct
58 Incorrect 51 ms 5976 KB Output isn't correct
59 Correct 78 ms 5992 KB Output is correct
60 Incorrect 51 ms 6232 KB Output isn't correct
61 Incorrect 58 ms 6232 KB Output isn't correct
62 Correct 113 ms 6236 KB Output is correct
63 Correct 70 ms 6236 KB Output is correct
64 Incorrect 132 ms 6236 KB Output isn't correct
65 Correct 112 ms 6236 KB Output is correct
66 Correct 92 ms 6228 KB Output is correct
67 Correct 78 ms 6232 KB Output is correct
68 Correct 37 ms 6096 KB Output is correct
69 Correct 34 ms 6236 KB Output is correct
70 Correct 32 ms 6224 KB Output is correct
71 Correct 36 ms 6232 KB Output is correct
72 Correct 26 ms 6096 KB Output is correct
73 Correct 39 ms 6480 KB Output is correct
74 Correct 43 ms 6492 KB Output is correct
75 Correct 50 ms 6488 KB Output is correct
76 Correct 45 ms 6488 KB Output is correct
77 Correct 56 ms 6488 KB Output is correct
78 Correct 56 ms 6492 KB Output is correct
79 Correct 46 ms 6324 KB Output is correct
80 Correct 49 ms 6492 KB Output is correct
81 Correct 57 ms 6476 KB Output is correct
82 Correct 49 ms 6492 KB Output is correct
83 Correct 58 ms 6236 KB Output is correct
84 Correct 59 ms 6224 KB Output is correct
85 Correct 58 ms 6432 KB Output is correct
86 Correct 64 ms 6228 KB Output is correct
87 Correct 55 ms 6232 KB Output is correct
88 Correct 71 ms 6400 KB Output is correct
89 Correct 63 ms 6232 KB Output is correct
90 Correct 71 ms 6236 KB Output is correct
91 Correct 67 ms 6236 KB Output is correct
92 Correct 65 ms 6232 KB Output is correct