TestCase/yukicoder/yuki0649.SortedMultiSetBIT.test.py

View this file on GitHub
Last update: 2021-02-06 18:04:37+09:00
Problem: https://yukicoder.me/problems/no/649

Depends on

順序付き多重集合 (Binary Indexed Tree) (DataStructure/SortedSet/SortedMultiSetBIT.py)

Code

# verification-helper: PROBLEM https://yukicoder.me/problems/no/649
import sys
input = sys.stdin.buffer.readline

from DataStructure.SortedSet.SortedMultiSetBIT import SortedMultiSetBIT


def main():
    # add, remove, kth_smallest, kth_largest
    q, k = map(int, input().split())
    queries = [list(map(int, input().split())) for i in range(q)]

    k -= 1
    cands = [query[0] for flag, *query in queries if flag == 1]
    sset = SortedMultiSetBIT(cands)

    ans = []
    for flag, *query in queries:
        if flag == 1:
            val = query[0]
            sset.add(val)
        else:
            if k < len(sset):
                val = sset.kth_smallest(k)
                assert(val == sset.kth_largest(len(sset) - k - 1))
                sset.remove(val)
                ans.append(val)
            else:
                ans.append(-1)

    print('\n'.join(map(str, ans)))

if __name__ == '__main__':
    main()

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.12.4/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.4/x64/lib/python3.12/site-packages/onlinejudge_verify/languages/python.py", line 96, in bundle
    raise NotImplementedError
NotImplementedError