src/$tests/misc/DeletableHeapq.test.py

View this file on GitHub
Last update: 2023-07-23 08:45:17+09:00
Problem: https://judge.yosupo.jp/problem/aplusb
Link: View error logs on GitHub Actions

Depends on

Code

# verification-helper: PROBLEM https://judge.yosupo.jp/problem/aplusb
from pathlib import Path
import sys

sys.path.append(str(Path(__file__).resolve().parent.parent.parent.parent))
from src.data_structure.BIT import BIT
from src.misc.DeletableHeapq import DeletableHeapq
import random

random.seed(10)

n = 500000
t = 500000
A = random.choices(range(n), k=t + 10)
cnt = len(A)
tot = sum(A)
if random.random() > 0.5:
    hq = DeletableHeapq(A)
else:
    hq = DeletableHeapq(A, reverse=True)
bit = BIT(n)
for a in A:
    bit.add(a, 1)

for _ in range(t):
    if random.random() < 0.3:
        p = hq[0]
        assert bit.get(p) >= 1
        if hq.pm == 1:
            assert bit.lower_bound(1) - 1 == p
        else:
            assert bit.lower_bound(cnt) - 1 == p
        bit.add(p, -1)
        assert p == hq.pop()
        cnt -= 1
        tot -= p
    else:
        p = random.randrange(n)
        if bit.get(p) == 0 or random.random() < 0.5:
            bit.add(p, 1)
            hq.push(p)
            cnt += 1
            tot += p
        else:
            bit.add(p, -1)
            assert hq.remove(p)
            cnt -= 1
            tot -= p

    assert len(hq) == cnt
    assert hq.tot == tot, (_, hq.tot, tot)


a, b = map(int, input().split())
print(a + b)

Traceback (most recent call last):
  File "/opt/hostedtoolcache/Python/3.11.4/x64/lib/python3.11/site-packages/onlinejudge_verify/documentation/build.py", line 81, in _render_source_code_stat
    bundled_code = language.bundle(
                   ^^^^^^^^^^^^^^^^
  File "/opt/hostedtoolcache/Python/3.11.4/x64/lib/python3.11/site-packages/onlinejudge_verify/languages/python.py", line 108, in bundle
    raise NotImplementedError
NotImplementedError