結果

問題 No.2977 Kth Xor Pair
コンテスト
ユーザー detteiuu
提出日時 2024-12-01 13:28:11
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,451 bytes
コンパイル時間 294 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 1,705,068 KB
最終ジャッジ日時 2024-12-01 13:30:18
合計ジャッジ時間 125,809 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 MLE * 1
other AC * 3 TLE * 3 MLE * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Trie:
    def __init__(self, N):
        self.tree = [[-1]*2 for _ in range(N+1)]
        self.arrive = [0]*(N+1)
        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[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(N*30)
    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