Submission #996987

# Submission time Handle Problem Language Result Execution time Memory
996987 2024-06-11T14:37:19 Z imann Mecho (IOI09_mecho) C++17
39 / 100
130 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 + (level[start.first][start.second] + 1) / s > 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;
}
# Verdict Execution time Memory Grader output
1 Correct 1 ms 2396 KB Output is correct
2 Correct 1 ms 2396 KB Output is correct
3 Correct 0 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 1 ms 2396 KB Output is correct
7 Correct 62 ms 6348 KB Output is correct
8 Correct 1 ms 2396 KB Output is correct
9 Correct 1 ms 2396 KB Output is correct
10 Correct 1 ms 2396 KB Output is correct
11 Correct 0 ms 2396 KB Output is correct
12 Incorrect 1 ms 4700 KB Output isn't correct
13 Incorrect 1 ms 2524 KB Output isn't correct
14 Correct 1 ms 4700 KB Output is correct
15 Incorrect 0 ms 2396 KB Output isn't correct
16 Incorrect 0 ms 2396 KB Output isn't correct
17 Incorrect 0 ms 2396 KB Output isn't correct
18 Incorrect 1 ms 2396 KB Output isn't correct
19 Incorrect 0 ms 2396 KB Output isn't correct
20 Incorrect 1 ms 2396 KB Output isn't correct
21 Incorrect 1 ms 2652 KB Output isn't correct
22 Incorrect 1 ms 2396 KB Output isn't correct
23 Incorrect 1 ms 2648 KB Output isn't correct
24 Incorrect 1 ms 2648 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 4696 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 4912 KB Output isn't correct
34 Incorrect 12 ms 4700 KB Output isn't correct
35 Correct 16 ms 4696 KB Output is correct
36 Incorrect 12 ms 4700 KB Output isn't correct
37 Incorrect 14 ms 4700 KB Output isn't correct
38 Correct 21 ms 4700 KB Output is correct
39 Incorrect 15 ms 4960 KB Output isn't correct
40 Incorrect 17 ms 4956 KB Output isn't correct
41 Correct 27 ms 4956 KB Output is correct
42 Incorrect 19 ms 5208 KB Output isn't correct
43 Incorrect 22 ms 5208 KB Output isn't correct
44 Correct 32 ms 5212 KB Output is correct
45 Incorrect 25 ms 5208 KB Output isn't correct
46 Incorrect 28 ms 5276 KB Output isn't correct
47 Correct 40 ms 5212 KB Output is correct
48 Incorrect 31 ms 5212 KB Output isn't correct
49 Incorrect 34 ms 5212 KB Output isn't correct
50 Correct 46 ms 5208 KB Output is correct
51 Incorrect 33 ms 5468 KB Output isn't correct
52 Incorrect 38 ms 5468 KB Output isn't correct
53 Correct 59 ms 5468 KB Output is correct
54 Incorrect 39 ms 5724 KB Output isn't correct
55 Incorrect 46 ms 5724 KB Output isn't correct
56 Correct 73 ms 5724 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 79 ms 5980 KB Output is correct
60 Incorrect 50 ms 6236 KB Output isn't correct
61 Incorrect 61 ms 6232 KB Output isn't correct
62 Correct 88 ms 6232 KB Output is correct
63 Correct 73 ms 6236 KB Output is correct
64 Incorrect 130 ms 6228 KB Output isn't correct
65 Correct 104 ms 6236 KB Output is correct
66 Correct 92 ms 6232 KB Output is correct
67 Correct 77 ms 6236 KB Output is correct
68 Correct 37 ms 6104 KB Output is correct
69 Correct 35 ms 6232 KB Output is correct
70 Correct 33 ms 6228 KB Output is correct
71 Correct 31 ms 6100 KB Output is correct
72 Incorrect 29 ms 6236 KB Output isn't correct
73 Incorrect 34 ms 6492 KB Output isn't correct
74 Correct 47 ms 6492 KB Output is correct
75 Correct 52 ms 6492 KB Output is correct
76 Correct 45 ms 6484 KB Output is correct
77 Correct 51 ms 6492 KB Output is correct
78 Correct 57 ms 6492 KB Output is correct
79 Correct 46 ms 6492 KB Output is correct
80 Correct 52 ms 6488 KB Output is correct
81 Correct 56 ms 6492 KB Output is correct
82 Correct 48 ms 6488 KB Output is correct
83 Correct 57 ms 6444 KB Output is correct
84 Correct 58 ms 6448 KB Output is correct
85 Correct 59 ms 6232 KB Output is correct
86 Correct 66 ms 6236 KB Output is correct
87 Correct 56 ms 6440 KB Output is correct
88 Correct 59 ms 6236 KB Output is correct
89 Correct 66 ms 6388 KB Output is correct
90 Correct 67 ms 6232 KB Output is correct
91 Correct 66 ms 6232 KB Output is correct
92 Correct 69 ms 6236 KB Output is correct