結果

問題 No.2786 RMQ on Grid Path
ユーザー soto800soto800
提出日時 2024-06-15 15:18:49
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 5,981 ms / 6,000 ms
コード長 7,588 bytes
コンパイル時間 3,526 ms
コンパイル使用メモリ 247,464 KB
実行使用メモリ 122,916 KB
最終ジャッジ日時 2024-06-15 15:20:08
合計ジャッジ時間 71,279 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,376 KB
testcase_02 AC 4 ms
5,376 KB
testcase_03 AC 5 ms
5,376 KB
testcase_04 AC 5 ms
5,376 KB
testcase_05 AC 5 ms
5,376 KB
testcase_06 AC 5 ms
5,376 KB
testcase_07 AC 5 ms
5,376 KB
testcase_08 AC 4 ms
5,376 KB
testcase_09 AC 4 ms
5,376 KB
testcase_10 AC 4 ms
5,376 KB
testcase_11 AC 4 ms
5,376 KB
testcase_12 AC 3,268 ms
122,916 KB
testcase_13 AC 3,251 ms
121,840 KB
testcase_14 AC 3,202 ms
122,604 KB
testcase_15 AC 3,205 ms
122,224 KB
testcase_16 AC 3,219 ms
122,632 KB
testcase_17 AC 3,222 ms
122,352 KB
testcase_18 AC 3,214 ms
121,964 KB
testcase_19 AC 3,244 ms
122,736 KB
testcase_20 AC 3,201 ms
122,224 KB
testcase_21 AC 3,197 ms
122,328 KB
testcase_22 AC 2,083 ms
119,028 KB
testcase_23 AC 2,098 ms
119,312 KB
testcase_24 AC 2,323 ms
119,920 KB
testcase_25 AC 2,406 ms
119,792 KB
testcase_26 AC 2,334 ms
119,924 KB
testcase_27 AC 1,122 ms
50,756 KB
testcase_28 AC 1,109 ms
51,840 KB
testcase_29 AC 2,515 ms
101,964 KB
testcase_30 AC 1,087 ms
47,876 KB
testcase_31 AC 108 ms
10,708 KB
testcase_32 AC 2,182 ms
117,096 KB
testcase_33 AC 5,981 ms
9,984 KB
testcase_34 AC 2,190 ms
117,232 KB
testcase_35 AC 2,220 ms
117,220 KB
testcase_36 AC 2,138 ms
117,104 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <cassert>

//#include <atcoder/all>
using namespace std;

#define lli int
#define REP(i, s, n) for (lli i = s; i < n; i++)
#define INF (1LL << 62)
#define mp(a, b) make_pair(a, b)
#define SORT(V) sort(V.begin(), V.end())
#define PI (3.141592653589794)
#define TO_STRING(VariableName) #VariableName
#define LOG1(x)                                                                \
  if (DEBUG)                                                                   \
    cout << TO_STRING(x) << "=" << x << " " << endl;
#define LOG2(x, y)                                                             \
  if (DEBUG)                                                                   \
    cout << TO_STRING(x) << "=" << x << " " << TO_STRING(y) << "=" << y << endl;
#define LOG3(x, y, z)                                                          \
  if (DEBUG)                                                                   \
    cout << TO_STRING(x) << "=" << x << " " << TO_STRING(y) << "=" << y << " " \
         << TO_STRING(z) << "=" << z << endl;
#define LOG4(w, x, y, z)                                                       \
  if (DEBUG)                                                                   \
    cout << TO_STRING(w) << "=" << w << " " << TO_STRING(x) << "=" << x << " " \
         << TO_STRING(y) << "=" << y << " " << TO_STRING(z) << "=" << z        \
         << endl;
#define LOG5(w, x, y, z, a)                                                    \
  if (DEBUG)                                                                   \
    cout << TO_STRING(w) << "=" << w << " " << TO_STRING(x) << "=" << x << " " \
         << TO_STRING(y) << "=" << y << " " << TO_STRING(z) << "=" << z << " " \
         << TO_STRING(a) << "=" << a << endl;
#define LOG6(w, x, y, z, a, b)                                                 \
  if (DEBUG)                                                                   \
    cout << TO_STRING(w) << "=" << w << " " << TO_STRING(x) << "=" << x << " " \
         << TO_STRING(y) << "=" << y << " " << TO_STRING(z) << "=" << z << " " \
         << TO_STRING(a) << "=" << a << " " << TO_STRING(b) << "=" << b        \
         << endl;

#define overload6(a, b, c, d, e, f, g, ...) g
#define LOG(...)                                                               \
  overload6(__VA_ARGS__, LOG6, LOG5, LOG4, LOG3, LOG2, LOG1)(__VA_ARGS__)

template <class T> bool chmax(T &a, const T &b) {
  if (a < b) {
    a = b;
    return 1;
  }
  return 0;
}
template <class T> bool chmin(T &a, const T &b) {
  if (b < a) {
    a = b;
    return 1;
  }
  return 0;
}

mt19937 engine;
std::chrono::system_clock::time_point start, endTime;

// checker https://atcoder.jp/contests/abc264/tasks/abc264_e

struct UnionFind {
  vector<int> par; // par[i]:iの親の番号　(例) par[3] = 2 : 3の親が2
  vector<bool> isOk;
  vector<lli> siz; // 素集合のサイズを表す配列(1 で初期化)
  lli N;

  UnionFind(int N)
      : par(N), isOk(N), siz(N) { //最初は全てが根であるとして初期化
    for (int i = 0; i < N; i++) {
      par[i] = i;
      isOk[i] = false;
      siz[i] = 1;
    }

    this->N = N;
  }

  int root(int x) { // データxが属する木の根を再帰で得る：root(x) = {xの木の根}
    if (par[x] == x)
      return x;
    return par[x] = root(par[x]);
  }

  void setOk(lli i) { isOk[i] = isOk[root(i)] = true; }

  bool isCheckOk(lli i) { return isOk[root(i)]; }
  void unite(int x, int y) { // xとyの木を併合
    int rx = root(x);        // xの根をrx
    int ry = root(y);        // yの根をry
    if (rx == ry)
      return; // xとyの根が同じ(=同じ木にある)時はそのまま

    if (siz[rx] < siz[ry])
      swap(rx, ry);
    siz[rx] += siz[ry];
    par[ry] =
        rx; // xとyの根が同じでない(=同じ木にない)時：xの根rxをyの根ryにつける
    if (isOk[rx] || isOk[ry]) {
      isOk[ry] = true;
      isOk[rx] = true;
    }
  }

  bool same(int x, int y) { // 2つのデータx, yが属する木が同じならtrueを返す
    int rx = root(x);
    int ry = root(y);
    return rx == ry;
  }
  lli size(lli x) { return siz[root(x)]; }
};

#define DEBUG 0

lli A[800][800];

lli di[2] = {1, 0};
lli dj[2] = {0, 1};

void solve() {
  //  write your code here

  lli H, W;
  cin >> H >> W;

  REP(i, 0, H) {
    REP(j, 0, W) { cin >> A[i][j]; }
  }

  using P = pair<lli, pair<pair<lli, lli>, pair<lli, lli>>>;

  priority_queue<P, vector<P>, greater<P>> pq;
  REP(i, 0, H) {
    REP(j, 0, W) {
      REP(dir, 0, 2) {
        lli ni = i + di[dir];
        lli nj = j + dj[dir];
        if (ni >= H || nj >= W)
          continue;
        lli cost = max(A[i][j], A[ni][nj]);
        pq.push({cost, {{i, j}, {ni, nj}}});
      }
    }
  }

  UnionFind uf(H * W);

  //片方の点の相方を入れる
  map<pair<lli, lli>, vector<pair<lli, lli>>> mp;

  map<lli, vector<pair<lli, lli>>> mp2;

  //クエリ番号を出す
  map<pair<pair<lli, lli>, pair<lli, lli>>, vector<lli>> pairToQueryNum;

  lli Q;
  cin >> Q;
  vector<lli> ans(Q, 0);
  REP(i, 0, Q) {
    lli i1, j1, i2, j2;
    cin >> i1 >> j1 >> i2 >> j2;
    i1--;
    j1--;
    i2--;
    j2--;
    mp[{i1, j1}].push_back({i2, j2});
    mp[{i2, j2}].push_back({i1, j1});
    pairToQueryNum[{{i1, j1}, {i2, j2}}].push_back(i);
    pairToQueryNum[{{i2, j2}, {i1, j1}}].push_back(i);
  }

  REP(i, 0, H) {
    REP(j, 0, W) { mp2[uf.root(i * W + j)].push_back({i, j}); }
  }

  while (!pq.empty()) {
    auto now = pq.top();
    pq.pop();
    lli cost = now.first;
    lli i1 = now.second.first.first;
    lli j1 = now.second.first.second;
    lli i2 = now.second.second.first;
    lli j2 = now.second.second.second;
    if (uf.same(i1 * W + j1, i2 * W + j2))
      continue;

    LOG(cost, i1, j1, i2, j2);
    //小さいサイズの方のイベントを確認する
    // i1,j1の方が小さい
    if (uf.size({i1 * W + j1}) > uf.size({i2 * W + j2})) {
      swap(i1, i2);
      swap(j1, j2);
    }

    lli smallRoot = uf.root(i1 * W + j1);
    lli largeRoot = uf.root(i2 * W + j2);

    lli smallRootSize = mp2[smallRoot].size();
    lli largeRootSize = mp2[largeRoot].size();

    //追加予定の点の座標を持って置く
    vector<pair<lli, lli>> addV;
    for (auto p : mp2[smallRoot]) {
      lli ni = p.first;
      lli nj = p.second;
      addV.push_back({ni, nj});
    }

    uf.unite(i1 * W + j1, i2 * W + j2);

    //追加予定のものを追加していくときにクエリがあるか見ていく
    for (auto p : addV) {
      lli ni = p.first;
      lli nj = p.second;
      for (auto np : mp[{ni, nj}]) {
        lli nni = np.first;
        lli nnj = np.second;
        if (uf.same(ni * W + nj, nni * W + nnj)) {
          for (auto nowQueryNum : pairToQueryNum[{p, np}]) {
            if (ans[nowQueryNum] == 0) {
              ans[nowQueryNum] = cost;
            }
          }
        }
      }
      // largeに突っ込んでおく
      mp2[largeRoot].push_back({ni, nj});
    }

    lli nowRootNum = uf.root(i1 * W + j1);
    lli nowRootNumSize = mp2[nowRootNum].size();
    LOG(smallRootSize, largeRootSize, nowRootNumSize);

    swap(mp2[nowRootNum], mp2[largeRoot]);
  }

  for (auto e : ans) {
    cout << e << endl;
  }
}

// Generated by 2.13.0 https://github.com/kyuridenamida/atcoder-tools  (tips:
// You use the default template now. You can remove this line by using your
// custom template)
int main() {
  // Failed to predict input format

  lli N = 1;
  // cin >> N;
  while (N--)
    solve();

  return 0;
}
0