結果

問題 No.2731 Two Colors
ユーザー InTheBloomInTheBloom
提出日時 2024-04-19 22:04:37
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 452 ms / 3,000 ms
コード長 2,581 bytes
コンパイル時間 1,309 ms
コンパイル使用メモリ 106,520 KB
実行使用メモリ 9,948 KB
最終ジャッジ日時 2024-10-11 15:14:18
合計ジャッジ時間 8,544 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 452 ms
7,808 KB
testcase_01 AC 452 ms
7,808 KB
testcase_02 AC 451 ms
7,808 KB
testcase_03 AC 5 ms
5,248 KB
testcase_04 AC 22 ms
5,248 KB
testcase_05 AC 15 ms
5,248 KB
testcase_06 AC 59 ms
5,248 KB
testcase_07 AC 72 ms
5,248 KB
testcase_08 AC 10 ms
5,248 KB
testcase_09 AC 269 ms
7,388 KB
testcase_10 AC 5 ms
5,248 KB
testcase_11 AC 238 ms
9,948 KB
testcase_12 AC 267 ms
7,376 KB
testcase_13 AC 199 ms
7,776 KB
testcase_14 AC 116 ms
5,248 KB
testcase_15 AC 9 ms
5,248 KB
testcase_16 AC 97 ms
5,404 KB
testcase_17 AC 146 ms
6,036 KB
testcase_18 AC 25 ms
5,248 KB
testcase_19 AC 100 ms
5,592 KB
testcase_20 AC 178 ms
6,132 KB
testcase_21 AC 27 ms
5,248 KB
testcase_22 AC 267 ms
8,212 KB
testcase_23 AC 64 ms
5,248 KB
testcase_24 AC 31 ms
5,248 KB
testcase_25 AC 3 ms
5,248 KB
testcase_26 AC 220 ms
6,932 KB
testcase_27 AC 269 ms
8,328 KB
testcase_28 AC 163 ms
7,132 KB
testcase_29 AC 51 ms
5,248 KB
testcase_30 AC 95 ms
5,680 KB
testcase_31 AC 56 ms
5,248 KB
testcase_32 AC 316 ms
9,096 KB
testcase_33 AC 2 ms
5,248 KB
testcase_34 AC 2 ms
5,248 KB
testcase_35 AC 2 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <queue>

using namespace std;

constexpr int iINF = 1'000'000'000;

int main () {
    int H, W; cin >> H >> W;
    vector<vector<int>> A(H, vector<int>(W));
    for (int i = 0; i < H; i++) {
        for (int j = 0; j < W; j++) cin >> A[i][j];
    }

    // グリッド上のダイクストラ法に帰着

    priority_queue<pair<int, pair<int, int>>, vector<pair<int, pair<int, int>>>, greater<pair<int, pair<int, int>>>> pq1, pq2;
    vector<vector<bool>> vis1(H, vector<bool>(W, false)), vis2(H, vector<bool>(W, false));

    pq1.push(make_pair(A[0][0], make_pair(0, 0)));
    pq2.push(make_pair(A[H - 1][W - 1], make_pair(H - 1, W - 1)));

    const vector<pair<int, int>> dxy = {
        make_pair(0, 1),
        make_pair(0, -1),
        make_pair(1, 0),
        make_pair(-1, 0),
    };

    bool end = false;
    int ans = 0;

    while (true) {
        {
            int y, x, c;
            while (true) {
                auto h = pq1.top(); pq1.pop();
                c = h.first;
                y = h.second.first, x = h.second.second;

                if (vis1[y][x]) continue;
                break;
            }
            vis1[y][x] = true;
            ans++;

            for (auto d : dxy) {
                int ny = y + d.first, nx = x + d.second;
                if (!(0 <= ny && ny < H && 0 <= nx && nx < W)) continue;
                if (vis1[ny][nx]) continue;

                // 相手側がもう塗られている
                if (vis2[ny][nx]) {
                    end = true;
                    break;
                }

                pq1.push(make_pair(A[ny][nx], make_pair(ny, nx)));
            }
        }

        if (end) break;

        {
            int y, x, c;
            while (true) {
                auto h = pq2.top(); pq2.pop();
                c = h.first;
                y = h.second.first, x = h.second.second;

                if (vis2[y][x]) continue;
                break;
            }
            vis2[y][x] = true;
            ans++;

            for (auto d : dxy) {
                int ny = y + d.first, nx = x + d.second;
                if (!(0 <= ny && ny < H && 0 <= nx && nx < W)) continue;
                if (vis2[ny][nx]) continue;

                // 相手側がもう塗られている
                if (vis1[ny][nx]) {
                    end = true;
                    break;
                }

                pq2.push(make_pair(A[ny][nx], make_pair(ny, nx)));
            }
        }

        if (end) break;
    }

    cout << ans - 2 << "\n";
}
0