結果
問題 | No.3134 二分探索木 |
ユーザー |
|
提出日時 | 2025-05-02 21:51:01 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 158 ms / 2,000 ms |
コード長 | 4,934 bytes |
コンパイル時間 | 3,642 ms |
コンパイル使用メモリ | 285,452 KB |
実行使用メモリ | 45,456 KB |
最終ジャッジ日時 | 2025-05-02 21:51:08 |
合計ジャッジ時間 | 6,401 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 5 |
other | AC * 15 |
ソースコード
#include <bits/stdc++.h> // #include <atcoder/all> using namespace std; // using namespace atcoder; #define rep(i, a, n) for(int i = a; i < n; i++) #define rrep(i, a, n) for(int i = a; i >= n; i--) #define inr(l, x, r) (l <= x && x < r) #define ll long long #define ld long double // using mint = modint1000000007; // using mint = modint998244353; constexpr int IINF = 1001001001; constexpr ll INF = 1e18; template<class t,class u> void chmax(t&a,u b){if(a<b)a=b;} template<class t,class u> void chmin(t&a,u b){if(b<a)a=b;} template <class T, T (*op)(T, T), T (*e)()> class SegmentTree { int _n, size, log; vector<T> d; void update(int k) { d[k] = op(d[2 * k], d[2 * k + 1]); } public: SegmentTree() : SegmentTree(0) {} explicit SegmentTree(int n) : SegmentTree(vector<T>(n, e())) {} // explicit で明示的に型を指定する explicit SegmentTree(const vector<T> &v) : _n(int(v.size())) { // sizeは_nを超える最小の2のべき乗 size = 1; while(size < _n) size *= 2, log++; // log は木の高さ(sizeの桁数) log = 0; while (!(size & (1 << log))) log++; d = vector<T>(2*size, e()); for(int i = 0; i < _n; i++) d[size+i] = v[i]; for(int i = size-1; i >= 1; i--){ update(i); } } void set(int p, T x){ assert(0 <= p && p < _n); p += size; d[p] = x; for(int i = 1; i <= log; i++) update(p >> i); } void apply(int p, T x){ set(p, op(x, d[p+size])); } T get(int p) const { assert(0 <= p && p < _n); return d[p+size]; } T prod(int l, int r) const { assert(0 <= l && l <= r && r <= _n); T sml = e(), smr = e(); l += size; r += size; while(l < r){ if(l&1) sml = op(sml, d[l++]); if(r&1) smr = op(d[--r], smr); l >>= 1; r >>= 1; } return op(sml, smr); } T all_prod() const {return d[1]; }; // f(op(a[l], a[l + 1], ..., a[r - 1])) = trueとなる最大のr template <bool (*f)(T)> int max_right(int l) const { return max_right(l, [](T x) { return f(x); }); } template <class F> int max_right(int l, F f) const { assert(0 <= l && l <= _n); assert(f(e())); if (l == _n) return _n; l += size; T sm = e(); do { while (l % 2 == 0) l >>= 1; if (!f(op(sm, d[l]))) { while (l < size) { l = (2 * l); if (f(op(sm, d[l]))) { sm = op(sm, d[l]); l++; } } return l - size; } sm = op(sm, d[l]); l++; } while ((l & -l) != l); return _n; } // f(op(a[l], a[l + 1], ..., a[r - 1])) = true となる最小の l template <bool (*f)(T)> int min_left(int r) const { return min_left(r, [](T x) { return f(x); }); } template <class F> int min_left(int r, F f) const { assert(0 <= r && r <= _n); assert(f(e())); if (r == 0) return 0; r += size; T sm = e(); do { r--; while (r > 1 && (r % 2)) r >>= 1; if (!f(op(d[r], sm))) { while (r < size) { r = (2 * r + 1); if (f(op(d[r], sm))) { sm = op(d[r], sm); r--; } } return r + 1 - size; } sm = op(d[r], sm); } while ((r & -r) != r); return 0; } }; using S = pair<int, int>; S op(S a, S b) { return min(a, b); } S e() { return {IINF, IINF}; } int main(){ int n; cin >> n; vector<int> a(n); rep(i, 0, n) cin >> a[i], a[i]--; SegmentTree<S, op, e> st(n); rep(i, 0, n) st.set(a[i], {i, a[i]}); vector<vector<int>> g(n); auto f = [&](auto self, int l, int r, int par) -> void { // cout << l << ' ' << r << ' ' << par << endl; { auto [val, id] = st.prod(l, par); if(val != IINF){ g[par].push_back(id); self(self, l, par, id); } } { auto [val, id] = st.prod(par+1, r); if(val != IINF){ g[par].push_back(id); self(self, par+1, r, id); } } }; f(f, 0, n, a[0]); vector<int> b(n), c(n); auto f2 = [&](auto self, int pos, int d) -> void { b[pos] = d; for(auto nxt: g[pos]){ self(self, nxt, d+1); c[pos] += c[nxt]+1; } }; f2(f2, a[0], 0); rep(i, 0, n) cout << b[a[i]] << ' '; cout << endl; rep(i, 0, n) cout << c[a[i]] << ' '; cout << endl; return 0; }