結果

問題 No.2977 Kth Xor Pair
ユーザー amesyuamesyu
提出日時 2024-12-01 02:10:18
言語 C++17(gcc12)
(gcc 12.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,316 bytes
コンパイル時間 842 ms
コンパイル使用メモリ 78,180 KB
実行使用メモリ 42,472 KB
最終ジャッジ日時 2024-12-01 02:11:28
合計ジャッジ時間 57,568 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,820 KB
testcase_02 AC 6 ms
6,816 KB
testcase_03 AC 6 ms
6,816 KB
testcase_04 AC 6 ms
6,816 KB
testcase_05 AC 7 ms
6,820 KB
testcase_06 AC 6 ms
6,816 KB
testcase_07 AC 1,332 ms
24,088 KB
testcase_08 AC 2,811 ms
41,132 KB
testcase_09 AC 2,416 ms
38,940 KB
testcase_10 AC 2,565 ms
38,072 KB
testcase_11 AC 2,625 ms
39,792 KB
testcase_12 AC 2,396 ms
36,908 KB
testcase_13 AC 2,625 ms
42,472 KB
testcase_14 AC 1,209 ms
22,152 KB
testcase_15 AC 2,200 ms
36,856 KB
testcase_16 AC 2,597 ms
40,740 KB
testcase_17 AC 2,659 ms
40,100 KB
testcase_18 AC 2,587 ms
41,112 KB
testcase_19 AC 2,628 ms
39,700 KB
testcase_20 AC 2,899 ms
39,100 KB
testcase_21 AC 2,657 ms
38,320 KB
testcase_22 TLE -
testcase_23 AC 2,687 ms
38,376 KB
testcase_24 AC 2,617 ms
39,436 KB
testcase_25 AC 2,465 ms
41,164 KB
testcase_26 AC 2,703 ms
38,992 KB
testcase_27 AC 874 ms
5,612 KB
testcase_28 AC 462 ms
5,612 KB
testcase_29 AC 519 ms
5,616 KB
testcase_30 AC 711 ms
5,732 KB
testcase_31 AC 705 ms
5,612 KB
testcase_32 AC 405 ms
5,248 KB
testcase_33 AC 397 ms
5,248 KB
testcase_34 AC 407 ms
5,248 KB
testcase_35 AC 416 ms
6,820 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

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

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

	void add_value(int x) {
		int root = 0;
		cnt[root]++;
		for(int i = log - 1; i >= 0; i--) {
			root = move(root, (x>>i)&1);
			cnt[root]++;
		}
	}
	
	inline int subtree_cnt(int r) {
		if(r == -1) return 0;
		return cnt[r];
	}

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

int main() {
	
	BinaryTrie bt;
	
	long long 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
	const long long pos = n + 2 * k - 1;
	int ng = -1, ok = (1<<log)-1;
	int x; long long count;
	while(ok - ng > 1) {
		x = (ok + ng) >> 1; count = 0;
		for(int i = 0; i < n; ++i) {
			count += bt.upper_bound(x, a[i]);
		}
		if(count >= pos) ok = x;
		else ng = x;
	}
	cout << ok << endl;
}
0