結果

問題 No.2977 Kth Xor Pair
ユーザー detteiuu
提出日時 2024-12-01 13:32:23
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,499 bytes
コンパイル時間 351 ms
コンパイル使用メモリ 82,228 KB
実行使用メモリ 782,664 KB
最終ジャッジ日時 2024-12-01 13:34:24
合計ジャッジ時間 119,726 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 3 TLE * 13 MLE * 18
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Trie:
    def __init__(self):
        self.tree = [[-1]*2]
        self.arrive = [0]
        self.nex = 1
    
    def insert(self, S):
        now = 0
        for i in range(30):
            b = 1&S>>(29-i)
            if self.tree[now][b] == -1:
                self.tree.append([-1, -1])
                self.arrive.append(0)
                self.tree[now][b] = self.nex
                now = self.nex
                self.nex += 1
            else:
                now = self.tree[now][b]
            self.arrive[now] += 1
    
    def query(self, MAX, S):
        now = 0
        ans = 0
        xor = MAX^S
        for i in range(30):
            b = 1 & S>>(29-i)
            if 1<<(29-i) & MAX:
                if b == 0 and self.tree[now][0] != -1:
                    ans += self.arrive[self.tree[now][0]]
                elif b == 1 and self.tree[now][1] != -1:
                    ans += self.arrive[self.tree[now][1]]
            b2 = 1 & xor>>(29-i)
            if self.tree[now][b2] != -1:
                now = self.tree[now][b2]
            else:
                break
        return ans

N, K = map(int, input().split())
A = list(map(int, input().split()))

def func(n):
    T = Trie()
    ans = 0
    for i in reversed(range(N)):
        T.insert(A[i])
        ans += T.query(n, A[i])-1
    del T
    return ans

left = 0
right = (1<<30)-1
while left+1 < right:
    mid = (left+right)//2
    if func(mid) < K:
        left = mid
    else:
        right = mid

print(right-1)
0