結果

問題 No.3134 二分探索木
コンテスト
ユーザー tobbie
提出日時 2025-05-31 00:54:17
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,578 bytes
コンパイル時間 2,099 ms
コンパイル使用メモリ 198,832 KB
実行使用メモリ 31,488 KB
最終ジャッジ日時 2025-05-31 00:54:25
合計ジャッジ時間 8,027 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 10 TLE * 1 -- * 4
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

#define rep(i, n) for (int i = 0; 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) {}
};

void insert(vector<struct Node> &node, int a, int ml, int mr, int x) {
  if (x < ml) {
    node[ml].l = x;
    node[x].p = ml;
    return;
  }
  if (x > mr) {
    node[mr].r = x;
    node[x].p = mr;
    return;
  }
  if (x == a) {
    node[x].p = 0;
    return;
  }
  while (1) {
    if (x < a) {
      if (node[a].l == -1) {
	node[a].l = x;
	node[x].p = a;
	break;
      }
      node[x].p = a;
      a = node[a].l;
      continue;
    }
    if (x > a) {
      if (node[a].r == -1) {
	node[a].r = x;
	node[x].p = a;
	break;
      }
      node[x].p = a;
      a = node[a].r;
    }
  }
}

int main() {
  int n;
  cin >> n;
  vector<struct Node> node(n, {-1, -1, -1});
  vector<int> a(n), ainv(n);
  int ml, mr;
  rep(i, n) {
    cin >> a[i]; a[i]--;
    if (i == 0) { ml = a[0]; mr = a[0]; }
    insert(node, a[0], ml, mr, a[i]);
    ainv[a[i]] = i;
    ml = min(a[i], ml);
    mr = max(a[i], mr);
  }
  vector<int> b(n), c(n);
  auto dfs = [&](auto dfs, int a, int d) {
    if (a == -1)
      return 0;
    int r = 0;
    b[ainv[a]] = d;
    r += dfs(dfs, node[a].l, d + 1);
    r += dfs(dfs, node[a].r, d + 1);
    c[ainv[a]] = 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