結果

問題 No.2786 RMQ on Grid Path
ユーザー eve__fuyukieve__fuyuki
提出日時 2024-06-14 22:47:30
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 3,453 ms / 6,000 ms
コード長 3,107 bytes
コンパイル時間 3,398 ms
コンパイル使用メモリ 241,288 KB
実行使用メモリ 59,984 KB
最終ジャッジ日時 2024-06-14 22:48:39
合計ジャッジ時間 57,925 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 2 ms
5,376 KB
testcase_03 AC 3 ms
5,376 KB
testcase_04 AC 2 ms
5,376 KB
testcase_05 AC 3 ms
5,376 KB
testcase_06 AC 2 ms
5,376 KB
testcase_07 AC 3 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 2 ms
5,376 KB
testcase_10 AC 3 ms
5,376 KB
testcase_11 AC 2 ms
5,376 KB
testcase_12 AC 3,345 ms
59,960 KB
testcase_13 AC 3,300 ms
59,984 KB
testcase_14 AC 3,355 ms
59,980 KB
testcase_15 AC 3,383 ms
59,980 KB
testcase_16 AC 3,453 ms
59,852 KB
testcase_17 AC 3,448 ms
59,980 KB
testcase_18 AC 3,341 ms
59,944 KB
testcase_19 AC 3,369 ms
59,960 KB
testcase_20 AC 3,365 ms
59,980 KB
testcase_21 AC 3,294 ms
59,896 KB
testcase_22 AC 1,109 ms
59,800 KB
testcase_23 AC 1,187 ms
59,788 KB
testcase_24 AC 1,602 ms
54,600 KB
testcase_25 AC 1,617 ms
54,676 KB
testcase_26 AC 1,604 ms
54,604 KB
testcase_27 AC 902 ms
17,372 KB
testcase_28 AC 613 ms
11,300 KB
testcase_29 AC 2,289 ms
55,196 KB
testcase_30 AC 666 ms
12,304 KB
testcase_31 AC 66 ms
5,972 KB
testcase_32 AC 800 ms
51,776 KB
testcase_33 AC 74 ms
10,048 KB
testcase_34 AC 1,624 ms
58,148 KB
testcase_35 AC 1,624 ms
58,188 KB
testcase_36 AC 1,633 ms
57,996 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

void fast_io() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
}

class UnionFind {
   public:
    vector<map<int, int>> par_or_rank;
    void init(int n) {
        par_or_rank.resize(n);
        for (int i = 0; i < n; i++) {
            par_or_rank[i][0] = -1;
        }
    };
    int leader(int u, int t = 1e9) {
        auto it = par_or_rank[u].lower_bound(t + 1);
        it--;
        int p = it->second;
        if (p < 0) {
            return u;
        }
        return leader(p, t);
    }
    bool same(int u, int v, int t = 1e9) {
        return leader(u, t) == leader(v, t);
    }
    void merge(int u, int v, int t) {
        u = leader(u, 1e9);
        v = leader(v, 1e9);
        if (u == v) {
            return;
        }
        if (-par_or_rank[u].rbegin()->second <
            -par_or_rank[v].rbegin()->second) {
            swap(u, v);
        }
        par_or_rank[u][t] =
            par_or_rank[u].rbegin()->second + par_or_rank[v].rbegin()->second;
        par_or_rank[v][t] = u;
    }
};
int main() {
    fast_io();
    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];
        }
    }
    vector<tuple<int, int, int>> edges;
    for (int i = 0; i < h; i++) {
        for (int j = 0; j < w; j++) {
            int id = i * w + j;
            if (i + 1 < h) {
                int id2 = (i + 1) * w + j;
                int cost = max(a[i][j], a[i + 1][j]);
                edges.emplace_back(cost, id, id2);
            }
            if (j + 1 < w) {
                int id2 = i * w + j + 1;
                int cost = max(a[i][j], a[i][j + 1]);
                edges.emplace_back(cost, id, id2);
            }
        }
    }
    sort(edges.begin(), edges.end());
    UnionFind uf;
    uf.init(h * w);
    for (auto [cost, u, v] : edges) {
        uf.merge(u, v, cost);
    }
    int q;
    cin >> q;
    vector<int> rs(q), cs(q), rt(q), ct(q);
    for (int i = 0; i < q; i++) {
        cin >> rs[i] >> cs[i] >> rt[i] >> ct[i];
        rs[i]--;
        cs[i]--;
        rt[i]--;
        ct[i]--;
    }
    vector<int> ans(q);
    map<pair<int, int>, vector<int>> mp;
    for (int i = 0; i < q; i++) {
        mp[{0, h * w}].push_back(i);
    }
    while (!mp.empty()) {
        auto [l, r] = mp.begin()->first;
        auto qs = mp.begin()->second;
        mp.erase(mp.begin());
        if (l + 1 == r) {
            for (int i : qs) {
                ans[i] = r;
            }
            continue;
        }
        int mid = (l + r) / 2;
        vector<int> lq, rq;
        for (int i : qs) {
            if (uf.same(rs[i] * w + cs[i], rt[i] * w + ct[i], mid)) {
                lq.push_back(i);
            } else {
                rq.push_back(i);
            }
        }
        if (!lq.empty()) {
            mp[{l, mid}] = lq;
        }
        if (!rq.empty()) {
            mp[{mid, r}] = rq;
        }
    }
    for (int i = 0; i < q; i++) {
        cout << ans[i] << "\n";
    }
}
0