algo — C++23 日本語リファレンス GitHub

test/yosupo/data_structure/predecessor_problem.test.cpp

AC · 最終更新 2026-08-01 · GitHub

問題: https://judge.yosupo.jp/problem/predecessor_problem

依存するヘッダ(1)
実装
// competitive-verifier: PROBLEM https://judge.yosupo.jp/problem/predecessor_problem
#include <iostream>
#include <string>
#include "data_structure/tree64.hpp"

int main(void) {
    int n, q;
    std::cin >> n >> q;
    std::string t;
    std::cin >> t;
    Tree64<10000000> st;
    for (int i = 0; i < n; ++i) {
        if (t[i] == '1') st.insert(i);
    }
    while (q--) {
        int c, k;
        std::cin >> c >> k;
        if (c == 0) st.insert(k);
        else if (c == 1) st.erase(k);
        else if (c == 2) std::cout << st.contains(k) << '\n';
        else if (c == 3) std::cout << st.succ(k) << '\n';
        else std::cout << st.pred(k + 1) << '\n';
    }

    return 0;
}