結果
| 問題 | No.649 ここでちょっとQK！ |
| ユーザー |
 Kohno
|
| 提出日時 | 2022-07-24 22:25:42 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 598 ms / 3,000 ms |
| コード長 | 5,590 bytes |
| コンパイル時間 | 615 ms |
| コンパイル使用メモリ | 86,876 KB |
| 実行使用メモリ | 190,824 KB |
| 最終ジャッジ日時 | 2023-09-20 22:57:26 |
| 合計ジャッジ時間 | 12,960 ms |
|
ジャッジサーバーID （参考情報） |
judge11 / judge12 |
（要ログイン）
テストケース
テストケース表示| 入力 | 結果 | 実行時間 実行使用メモリ |
|---|---|---|
| testcase_00 | AC | 77 ms
71,564 KB |
| testcase_01 | AC | 77 ms
71,300 KB |
| testcase_02 | AC | 75 ms
71,232 KB |
| testcase_03 | AC | 263 ms
98,220 KB |
| testcase_04 | AC | 474 ms
190,824 KB |
| testcase_05 | AC | 462 ms
190,412 KB |
| testcase_06 | AC | 355 ms
115,984 KB |
| testcase_07 | AC | 74 ms
71,572 KB |
| testcase_08 | AC | 75 ms
71,412 KB |
| testcase_09 | AC | 76 ms
71,368 KB |
| testcase_10 | AC | 75 ms
71,664 KB |
| testcase_11 | AC | 76 ms
71,468 KB |
| testcase_12 | AC | 367 ms
124,680 KB |
| testcase_13 | AC | 354 ms
124,472 KB |
| testcase_14 | AC | 347 ms
124,768 KB |
| testcase_15 | AC | 365 ms
124,636 KB |
| testcase_16 | AC | 354 ms
125,104 KB |
| testcase_17 | AC | 376 ms
114,524 KB |
| testcase_18 | AC | 406 ms
137,740 KB |
| testcase_19 | AC | 425 ms
143,772 KB |
| testcase_20 | AC | 445 ms
146,296 KB |
| testcase_21 | AC | 475 ms
155,964 KB |
| testcase_22 | AC | 506 ms
157,840 KB |
| testcase_23 | AC | 520 ms
166,824 KB |
| testcase_24 | AC | 532 ms
148,028 KB |
| testcase_25 | AC | 572 ms
178,172 KB |
| testcase_26 | AC | 598 ms
183,952 KB |
| testcase_27 | AC | 106 ms
77,576 KB |
| testcase_28 | AC | 104 ms
77,440 KB |
| testcase_29 | AC | 98 ms
76,384 KB |
| testcase_30 | AC | 323 ms
113,816 KB |
| testcase_31 | AC | 316 ms
114,308 KB |
| testcase_32 | AC | 77 ms
71,268 KB |
| testcase_33 | AC | 74 ms
71,604 KB |
| testcase_34 | AC | 75 ms
71,628 KB |
| testcase_35 | AC | 75 ms
71,512 KB |
ソースコード
class SegmentTree(object):
def __init__(self, v, op, e):
self._n = len(v)
self.op = op
self.e = e
self.log = SegmentTree._ceil_pow2(self._n)
self.size = 1 << self.log
self.d = [self.e] * 2 * self.size
for i in range(self._n):
self.d[self.size + i] = v[i]
for i in range(self.size - 1, 0, -1):
self._update(i)
@staticmethod
def _ceil_pow2(n):
for x in range(64):
if 1 << x >= n:
break
return x
def _update(self, k):
self.d[k] = self.op(self.d[2 * k], self.d[2 * k + 1])
def __getitem__(self, p):
assert 0 <= p < self._n
return self.d[p + self.size]
def __setitem__(self, p, x):
assert 0 <= p < self._n
p += self.size
self.d[p] = x
for i in range(1, self.log + 1):
self._update(p >> i)
def __str__(self):
return str(self.d[self.size:])
def query(self, l, r):
if l is ...:
l = 0
if r is ...:
r = self._n
assert 0 <= l <= r <= self._n
sml = self.e
smr = self.e
l += self.size
r += self.size
while l < r:
if l & 1:
sml = self.op(sml, self.d[l])
l += 1
if r & 1:
r -= 1
smr = self.op(self.d[r], smr)
l >>= 1
r >>= 1
return self.op(sml, smr)
def max_right(self, l, f):
if l is ...:
l = 0
assert 0 <= l <= self._n
assert f(self.e)
if l == self._n:
return self._n
l += self.size
sm = self.e
while True:
while l & 1 == 0:
l >>= 1
if not f(self.op(sm, self.d[l])):
while l < self.size:
l <<= 1
if f(self.op(sm, self.d[l])):
sm = self.op(sm, self.d[l])
l += 1
return l - self.size
sm = self.op(sm, self.d[l])
l += 1
if (l & -l) == l:
break
return self._n
def min_left(self, r, f):
if r is ...:
r = self._n
assert 0 <= r <= self._n
assert f(self.e)
if r == 0:
return 0
r += self.size
sm = self.e
while True:
r -= 1
while r > 1 and r & 1:
r >>= 1
if not f(self.op(self.d[r], sm)):
while r < self.size:
r <<= 1
r += 1
if f(self.op(self.d[r], sm)):
sm = self.op(self.d[r], sm)
r -= 1
return r + 1 - self.size
sm = self.op(self.d[r], sm)
if (r & -r) == r:
break
return 0
class _CountOrItem(object):
def __init__(self, _tree_map):
self._tree = _tree_map._tree
self._g = _tree_map._g
def _bind(self, l, r):
self._l = l
self._r = r
def _predicate(self, s):
return s < self._tree.query(self._l, self._r)
def count(self):
return self._tree.query(self._l, self._r)
def item(self):
try:
if self._l is ...:
i = self._tree.max_right(..., self._predicate)
else:
i = self._tree.min_left(..., self._predicate) - 1
except AssertionError:
return None
return self._g[i] if 0 <= i < len(self._g) else None
from operator import add
class LimitedTreeMap(object):
def __init__(self, domain):
self._g = sorted(set(domain))
self._f = {v: k for k, v in enumerate(self._g)}
self._tree = SegmentTree([0] * len(self._g), add, 0)
self._count_or_item = _CountOrItem(self)
def __setitem__(self, p, x):
p = self._f[p]
self._tree[p] = x
def __getitem__(self, p):
p = self._f[p]
return self._tree[p]
def __str__(self):
return str({v: self._tree[k] for v, k in self._f.items() if self._tree[k]})
def __repr__(self):
return str({v: self._tree[k] for v, k in self._f.items() if self._tree[k]})
def __lt__(self, p):
p = self._f[p]
self._count_or_item._bind(..., p)
return self._count_or_item
def __le__(self, p):
p = self._f[p]
self._count_or_item._bind(..., p + 1)
return self._count_or_item
def __gt__(self, p):
p = self._f[p]
self._count_or_item._bind(p + 1, ...)
return self._count_or_item
def __ge__(self, p):
p = self._f[p]
self._count_or_item._bind(p, ...)
return self._count_or_item
def k_smallest(self, k):
i = self._tree.max_right(..., lambda s: s < k + 1)
return self._g[i] if 0 <= i < len(self._g) else None
def k_largest(self, k):
i = self._tree.min_left(..., lambda s: s < k + 1) - 1
return self._g[i] if 0 <= i < len(self._g) else None
q, k = map(int, input().split())
domain = []
queries = []
for _ in range(q):
t, *args = map(int, input().split())
if t == 1:
domain.append(*args)
queries.append((t, *args))
tree_map = LimitedTreeMap(domain)
for t, *args in queries:
if t == 1:
x, = args
tree_map[x] += 1
else:
x = tree_map.k_smallest(k - 1)
if x is None:
x = -1
else:
tree_map[x] -= 1
print(x)