#1029214 (C++23) No.2977 Kth Xor Pair

• 問題: No.2977 Kth Xor Pair
• ユーザー: amesyu
• 提出日時: 2024-12-01 01:17:23
• 言語: C++23 (gcc 13.3.0 + boost 1.87.0)
• 結果: MLE
• 実行時間: -
• コード長: 2,036 bytes
• コンパイル時間: 1,199 ms
• コンパイル使用メモリ: 92,700 KB
• 実行使用メモリ: 541,712 KB
• 最終ジャッジ日時: 2024-12-01 01:18:06
• 合計ジャッジ時間: 42,258 ms
• ジャッジサーバーID （参考情報）: judge4 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 13,632 KB
testcase_01	MLE	-
testcase_02	AC	21 ms; 18,732 KB
testcase_03	MLE	-
testcase_04	AC	22 ms; 19,772 KB
testcase_05	MLE	-
testcase_06	AC	23 ms; 19,956 KB
testcase_07	MLE	-
testcase_08	TLE	-
testcase_09	MLE	-
testcase_10	WA	-
testcase_11	WA	-
testcase_12	WA	-
testcase_13	WA	-
testcase_14	WA	-
testcase_15	WA	-
testcase_16	WA	-
testcase_17	WA	-
testcase_18	WA	-
testcase_19	TLE	-
testcase_20	WA	-
testcase_21	WA	-
testcase_22	TLE	-
testcase_23	TLE	-
testcase_24	TLE	-
testcase_25	WA	-
testcase_26	TLE	-
testcase_27	WA	-
testcase_28	WA	-
testcase_29	WA	-
testcase_30	WA	-
testcase_31	WA	-
testcase_32	WA	-
testcase_33	WA	-
testcase_34	WA	-
testcase_35	MLE	-

ソースコード

#include <iostream>
#include <vector>
using namespace std;
using ll = long long;
const int log = 30;
struct BinaryTrie {
	
	struct Node { int par, cnt, b0, b1; };
	vector<Node> arr;
	
	explicit BinaryTrie(void) {
		arr.push_back(Node{-1, 0, -1, -1});
	}

	int __append(int p, int nxt_bit) {
		int idx = arr.size();
		arr.push_back({p, 0, -1, -1});
		if(nxt_bit == 0) arr[p].b0 = idx;
		if(nxt_bit == 1) arr[p].b1 = idx;
		return idx;
	}

	int move(int p, int nxt_bit) {
		if(nxt_bit == 0) {
			if(arr[p].b0 == -1) return __append(p, nxt_bit);
			else return arr[p].b0;
		} else {
			if(arr[p].b1 == -1) return __append(p, nxt_bit);
			else return arr[p].b1;
		}
	}

	void add_value(int x) {
		int root = 0;
		for(int i = log; i >= 0; i--) {
			root = move(root, (x>>i)&1);
		}
		while(root >= 0) {
			arr[root].cnt++;
			root = arr[root].par;
		}
	}
	
	int subtree_cnt(int r) {
		if(r == -1) return 0;
		return arr[r].cnt;
	}
	int lower_bound(int value, int x=0) {
		int lb = 0;
		int root = 0;
		for(int i = log; i >= 0; i--) {
			int nxt_bit = (value>>i)&1;
			int flip = (x>>i)&1;
			if(nxt_bit == 1) {
				if(flip) lb += subtree_cnt(arr[root].b1);
				else lb += subtree_cnt(arr[root].b0);
			}
			root = move(root, nxt_bit^flip);
		}
		return lb;
	}
	
	int upper_bound(int value, int x=0) {
		int ub = 0;
		int root = 0;
		for(int i = log; i >= 0; i--) {
			int nxt_bit = (value>>i)&1;
			int flip = (x>>i)&1;
			if(nxt_bit == 1) {
				if(flip) ub += subtree_cnt(arr[root].b1);
				else ub += subtree_cnt(arr[root].b0);
			}
			root = move(root, nxt_bit^flip);
		}
		ub += subtree_cnt(root);
		return ub;
	}
};

int main() {
	
	BinaryTrie bt;
	
	int n, k;
	cin >> n >> k;
	
	vector<int> a(n);
	for(int i = 0; i < n; ++i) {
		cin >> a[i];
		bt.add_value(a[i]);
	}
		
	// Find N + 2 * K kth number
	int ng = -1, ok = 1e9;
	while(ok - ng > 1) {
		int x = (ok + ng) >> 1;
		int count = 0;
		for(int i = 0; i < n; ++i) {
			count += bt.upper_bound(x, a[i]);
		}
		if(count >= n + 2 * k) ok = x;
		else ng = x;
	}
	cout << ok << endl;
}