結果

問題 No.3134 二分探索木
ユーザー tobbie
提出日時 2025-05-31 16:04:58
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 112 ms / 2,000 ms
コード長 1,399 bytes
コンパイル時間 2,314 ms
コンパイル使用メモリ 197,672 KB
実行使用メモリ 31,488 KB
最終ジャッジ日時 2025-05-31 16:05:04
合計ジャッジ時間 4,943 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

#define rep(i, n) for (int i = 0; i < (int)n; i++)
#define rep1(i, j, n) for (int i = (int)j; i < (int)n; i++)

struct Node {
  int p;
  int l;
  int r;
  Node (int p, int l, int r) : p(p), l(l), r(r) {}
};

int main() {
  int n;
  cin >> n;
  vector<struct Node> node(n, {-1, -1, -1});
  vector<int> a(n), ainv(n+1, -1);
  rep(i, n) {
    cin >> a[i]; a[i]--;
    ainv[a[i]] = i;
  }
  rep(i, n-1) {
    if (ainv[i] < ainv[i+1]) {
      node[i+1].p = i;
    }
    else {
      int j = i;
      while (1) {
	if (node[j].p == -1) {
	  node[j].p = i+1;
	  break;
	} else if (ainv[node[j].p] < ainv[i+1]) {
	  node[i+1].p = node[j].p;
	  node[j].p = i+1;
	  break;
	} else {
	  j = node[j].p;
	}
      }
    }
  }
  rep(i, n) {
    if (node[i].p == -1)
      continue;
    if (i < node[i].p)
      node[node[i].p].l = i;
    else
      node[node[i].p].r = i;
  }
  vector<int> b(n), c(n);
  auto dfs = [&](auto dfs, int a, int d) {
    if (a == -1)
      return 0;
    int i = ainv[a];
    b[i] = d;
    int r = 0;
    r += dfs(dfs, node[a].l, d + 1);
    r += dfs(dfs, node[a].r, d + 1);
    c[i] = r;
    return r + 1;
  };
  dfs(dfs, a[0], 0);
  rep(i, n) {
    cout << b[i];
    if (i < n-1) cout << " ";
    else         cout << endl;
  }
  rep(i, n) {
    cout << c[i];
    if (i < n-1) cout << " ";
    else         cout << endl;
  }
  return 0;
}
0