結果

問題 No.2977 Kth Xor Pair
ユーザー amesyuamesyu
提出日時 2024-12-20 03:05:29
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 2,466 bytes
コンパイル時間 165 ms
コンパイル使用メモリ 82,408 KB
実行使用メモリ 525,492 KB
最終ジャッジ日時 2024-12-20 03:07:26
合計ジャッジ時間 113,128 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 44 ms
66,400 KB
testcase_01 AC 45 ms
68,684 KB
testcase_02 AC 125 ms
85,324 KB
testcase_03 AC 123 ms
85,080 KB
testcase_04 AC 124 ms
84,460 KB
testcase_05 AC 121 ms
77,944 KB
testcase_06 AC 123 ms
84,632 KB
testcase_07 MLE -
testcase_08 TLE -
testcase_09 TLE -
testcase_10 TLE -
testcase_11 TLE -
testcase_12 TLE -
testcase_13 TLE -
testcase_14 TLE -
testcase_15 TLE -
testcase_16 TLE -
testcase_17 TLE -
testcase_18 TLE -
testcase_19 TLE -
testcase_20 TLE -
testcase_21 TLE -
testcase_22 TLE -
testcase_23 TLE -
testcase_24 TLE -
testcase_25 TLE -
testcase_26 TLE -
testcase_27 TLE -
testcase_28 TLE -
testcase_29 TLE -
testcase_30 TLE -
testcase_31 TLE -
testcase_32 AC 2,899 ms
102,160 KB
testcase_33 TLE -
testcase_34 AC 2,810 ms
101,956 KB
testcase_35 AC 2,594 ms
210,120 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

# Pythonは低速なので通るわけがない

class BinaryTrie:

    def __init__(self, log):
        self.lg = log
        self.b0 = [-1]
        self.b1 = [-1]
        self.cnt = [0]

    def __append(self, p, nxt_bit):
        idx = len(self.cnt)
        self.b0.append(-1)
        self.b1.append(-1)
        self.cnt.append(0)
        if nxt_bit: self.b1[p] = idx
        else: self.b0[p] = idx
        return idx

    def move(self, p, nxt_bit):
        if nxt_bit:
            if self.b1[p] == -1: return self.__append(p, nxt_bit)
            return self.b1[p]
        else:
            if self.b0[p] == -1: return self.__append(p, nxt_bit)
            return self.b0[p]
    
    def subtree_cnt(self, r):
        if r == -1: return 0
        return self.cnt[r]

    def add_value(self, x):
        r = 0
        self.cnt[r] += 1
        for i in range(self.lg, -1, -1):
            r = self.move(r, (x>>i)&1)
            self.cnt[r] += 1

    def lower_bound(self, value, x=0):
        lb = 0
        r = 0
        for i in range(self.lg, -1, -1):
            if self.cnt[r] == 0: return lb
            nxt_bit = (value>>i)&1
            flip = (x>>i)&1
            if nxt_bit:
                if flip: lb += self.subtree_cnt(self.b1[r])
                else: lb += self.subtree_cnt(self.b0[r])
            if nxt_bit^flip == 0 and self.b0[r] == -1: return lb
            if nxt_bit^flip == 1 and self.b1[r] == -1: return lb
            r = self.move(r, nxt_bit^flip)
        return lb
    
    def upper_bound(self, value, x=0):
        ub = 0
        r = 0
        for i in range(self.lg, -1, -1):
            if self.cnt[r] == 0: return ub
            nxt_bit = (value>>i)&1
            flip = (x>>i)&1
            if nxt_bit:
                if flip: ub += self.subtree_cnt(self.b1[r])
                else: ub += self.subtree_cnt(self.b0[r])
            if nxt_bit^flip == 0 and self.b0[r] == -1: return ub
            if nxt_bit^flip == 1 and self.b1[r] == -1: return ub
            r = self.move(r, nxt_bit^flip)
        ub += self.subtree_cnt(r)
        return ub

n, k = map(int, input().split())
bt = BinaryTrie(30)
a = list(map(int, input().split()))
for v in a: bt.add_value(v)

pos = n + 2 * k - 1

ng, ok = -1, (1 << 30) - 1
while ok - ng > 1:
    x = (ok + ng) >> 1
    count = 0

    for i in range(n):
        count += bt.upper_bound(x, a[i])
        if count >= pos: break
        
    if count >= pos: ok = x
    else: ng = x

print(ok)
0