verify/yosupo_ds/ds_ordered_set_max_right.test.cpp

• View this file on GitHub
• Last update: 2025-06-19 13:39:58+09:00
• Problem: https://judge.yosupo.jp/problem/ordered_set

Depends on

• 赤黒木の基本クラス (bbst/base/red_black_tree_base.hpp)
• bbst/red_black_tree.hpp
• 常に右側を保持するモノイド (math/monoid/hold.hpp)
• others/vector_pool.hpp
• template/constant.hpp
• template/fastio.hpp
• template/io_util.hpp
• template/macros.hpp
• template/template.hpp
• template/type_alias.hpp
• type_traits/integral.hpp
• type_traits/io.hpp

Code

#define PROBLEM "https://judge.yosupo.jp/problem/ordered_set" 

#include "../../bbst/red_black_tree.hpp"
#include "../../math/monoid/hold.hpp"
#include "../../template/template.hpp"
using namespace std;

int main() {
    int n, q;
    kin >> n >> q;
    vc<kk2::monoid::Hold<int>> a(n);
    kin >> a;
    kk2::RedBlackTree<kk2::monoid::Hold<int>> rb(2 * (n + q));
    auto root = rb.build(a);
    rep (q) {
        int t, x;
        kin >> t >> x;
        if (t == 0) {
            auto [k, a, b] = rb.max_right(root, 0, [&](auto t) { return t.is_unit or t.a < x; });
            if (b and b->val.a == x) continue;
            rb.insert(root, k, x);
        } 
        if (t == 1) {
            auto [k, a, b] = rb.max_right(root, 0, [&](auto t) { return t.is_unit or t.a < x; });
            if (b and b->val.a == x) rb.erase(root, k);
        }
        if (t == 2) {
            --x;
            if (rb.size(root) <= x) kout << -1 << "\n";
            else kout << rb.get(root, x) << "\n";
        }
        if (t == 3) {
            kout << rb.max_right(root, 0, [&](auto t) { return t.is_unit or t.a <= x; }).s << "\n";
        }
        if (t == 4) {
            auto [k, a, b] = rb.max_right(root, 0, [&](auto t) { return t.is_unit or t.a <= x; });
            if (a.is_unit) kout << -1 << "\n";
            else kout << a << "\n";
        }
        if (t == 5) {
            auto [k, a, b] = rb.max_right(root, 0, [&](auto t) { return t.is_unit or t.a < x; });
            if (b) kout << b->val << "\n";
            else kout << -1 << "\n";
        }
    }

    return 0;
}

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/documentation/build.py", line 71, in _render_source_code_stat
    bundled_code = language.bundle(stat.path, basedir=basedir, options={'include_paths': [basedir]}).decode()
                   ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/cplusplus.py", line 187, in bundle
    bundler.update(path)
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/cplusplus_bundle.py", line 401, in update
    self.update(self._resolve(pathlib.Path(included), included_from=path))
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/cplusplus_bundle.py", line 401, in update
    self.update(self._resolve(pathlib.Path(included), included_from=path))
  File "/opt/hostedtoolcache/Python/3.12.0/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/cplusplus_bundle.py", line 312, in update
    raise BundleErrorAt(path, i + 1, "#pragma once found in a non-first line")
onlinejudge_verify.languages.cplusplus_bundle.BundleErrorAt: type_traits/io.hpp: line 4: #pragma once found in a non-first line

Back to top page