結果
問題 | No.649 ここでちょっとQK! |
ユーザー | 👑 rin204 |
提出日時 | 2022-07-04 14:13:06 |
言語 | PyPy3 (7.3.13) |
結果 |
RE
|
実行時間 | - |
コード長 | 3,687 bytes |
コンパイル時間 | 558 ms |
コンパイル使用メモリ | 84,244 KB |
実行使用メモリ | 658,424 KB |
最終ジャッジ日時 | 2023-08-20 21:07:08 |
合計ジャッジ時間 | 32,258 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge12 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 59 ms
71,112 KB |
testcase_01 | AC | 59 ms
71,300 KB |
testcase_02 | AC | 59 ms
71,236 KB |
testcase_03 | AC | 418 ms
78,332 KB |
testcase_04 | AC | 372 ms
110,156 KB |
testcase_05 | AC | 373 ms
110,528 KB |
testcase_06 | AC | 303 ms
78,568 KB |
testcase_07 | AC | 61 ms
71,332 KB |
testcase_08 | AC | 61 ms
71,320 KB |
testcase_09 | RE | - |
testcase_10 | RE | - |
testcase_11 | RE | - |
testcase_12 | AC | 980 ms
346,792 KB |
testcase_13 | AC | 1,098 ms
351,184 KB |
testcase_14 | AC | 890 ms
339,260 KB |
testcase_15 | AC | 998 ms
371,720 KB |
testcase_16 | AC | 1,167 ms
394,336 KB |
testcase_17 | AC | 1,321 ms
416,996 KB |
testcase_18 | AC | 1,469 ms
443,880 KB |
testcase_19 | AC | 1,428 ms
475,652 KB |
testcase_20 | AC | 1,624 ms
509,112 KB |
testcase_21 | MLE | - |
testcase_22 | MLE | - |
testcase_23 | MLE | - |
testcase_24 | MLE | - |
testcase_25 | MLE | - |
testcase_26 | MLE | - |
testcase_27 | AC | 188 ms
76,964 KB |
testcase_28 | AC | 96 ms
77,016 KB |
testcase_29 | AC | 98 ms
77,012 KB |
testcase_30 | AC | 360 ms
89,112 KB |
testcase_31 | AC | 372 ms
91,788 KB |
testcase_32 | AC | 59 ms
70,952 KB |
testcase_33 | RE | - |
testcase_34 | AC | 61 ms
71,056 KB |
testcase_35 | AC | 66 ms
70,880 KB |
ソースコード
class Binary_Trie_Vertex: def __init__(self): self.left = None self.right = None self.parent = None self.size = 0 class Binary_Trie: def __init__(self, logn, ismulti = False): self.logn = logn self.root = Binary_Trie_Vertex() self.bi = [1 << i for i in range(logn)] self.multi = ismulti def add(self, x): V = self.root plus = False for i in range(self.logn - 1, -1, -1): if x >> i & 1: if V.right is None: V.right = Binary_Trie_Vertex() V.right.parent = V plus = True V = V.right else: if V.left is None: V.left = Binary_Trie_Vertex() V.left.parent = V plus = True V = V.left if plus or self.multi: while V.parent is not None: V.size += 1 V = V.parent V.size += 1 def remove(self, x): V = self.root for i in range(self.logn - 1, -1, -1): if x >> i & 1: if V.right is None: return False V = V.right else: if V.left is None: return False V = V.left V2 = V while V2.parent is not None: V2.size -= 1 V2 = V2.parent V2.size -= 1 while V.size == 0: P = V.parent if P.right == V: P.right = None else: P.left = None if P.right is None and P.left is None: pass else: return True V = P return True def search(self, x): V = self.root for i in range(self.logn - 1, -1, -1): if x >> i & 1: if V.right is None: return False V = V.right else: if V.left is None: return False V = V.left return True def xor_min(self, x): xor = 0 V = self.root for i in range(self.logn - 1, -1, -1): if x >> i & 1: if V.right is not None: V = V.right else: xor += self.bi[i] V = V.left else: if V.left is not None: V = V.left else: xor += self.bi[i] V = V.right return xor def __getitem__(self, k): return self.Kth_element(k) def Kth_element(self, k): if k < 0: k = self.root.size + k k += 1 if k > self.root.size or k <= 0: return None ret = 0 V = self.root for i in range(self.logn - 1, -1, -1): if V.left is None: ret += self.bi[i] V = V.right elif V.right is None: V = V.left elif V.left.size >= k: V = V.left else: ret += self.bi[i] k -= V.left.size V = V.right return ret Q, k = map(int, input().split()) bt = Binary_Trie(60, True) for _ in range(Q): query = list(map(int, input().split())) if query[0] == 1: bt.add(query[1]) else: ans = bt[k - 1] if ans is None: print(-1) else: print(ans) bt.remove(ans)