結果

問題 No.2786 RMQ on Grid Path
ユーザー eve__fuyuki
提出日時 2024-06-14 22:48:45
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,385 ms / 6,000 ms
コード長 2,512 bytes
コンパイル時間 2,427 ms
コンパイル使用メモリ 220,112 KB
最終ジャッジ日時 2025-02-21 22:17:49
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

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]--;
        int ng = 0, ok = h * w;
        while (ok - ng > 1) {
            int mid = (ok + ng) / 2;
            if (uf.same(rs[i] * w + cs[i], rt[i] * w + ct[i], mid)) {
                ok = mid;
            } else {
                ng = mid;
            }
        }
        cout << ok << "\n";
    }
}
0