結果

問題 No.2977 Kth Xor Pair
ユーザー detteiuudetteiuu
提出日時 2024-12-01 13:19:37
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,542 bytes
コンパイル時間 275 ms
コンパイル使用メモリ 82,444 KB
実行使用メモリ 1,698,116 KB
最終ジャッジ日時 2024-12-01 13:21:36
合計ジャッジ時間 107,169 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 52 ms
67,584 KB
testcase_01 MLE -
testcase_02 AC 204 ms
104,412 KB
testcase_03 MLE -
testcase_04 MLE -
testcase_05 AC 161 ms
105,080 KB
testcase_06 AC 162 ms
105,124 KB
testcase_07 MLE -
testcase_08 TLE -
testcase_09 MLE -
testcase_10 MLE -
testcase_11 MLE -
testcase_12 TLE -
testcase_13 TLE -
testcase_14 MLE -
testcase_15 MLE -
testcase_16 MLE -
testcase_17 TLE -
testcase_18 MLE -
testcase_19 MLE -
testcase_20 MLE -
testcase_21 MLE -
testcase_22 MLE -
testcase_23 MLE -
testcase_24 MLE -
testcase_25 MLE -
testcase_26 TLE -
testcase_27 TLE -
testcase_28 TLE -
testcase_29 TLE -
testcase_30 TLE -
testcase_31 TLE -
testcase_32 MLE -
testcase_33 MLE -
testcase_34 MLE -
testcase_35 MLE -
権限があれば一括ダウンロードができます

ソースコード

diff # 

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