結果

問題 No.2977 Kth Xor Pair
ユーザー amesyu
提出日時 2024-12-01 02:14:58
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,846 ms / 3,000 ms
コード長 2,380 bytes
コンパイル時間 1,030 ms
コンパイル使用メモリ 95,248 KB
実行使用メモリ 42,552 KB
最終ジャッジ日時 2024-12-01 02:15:59
合計ジャッジ時間 47,996 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 34
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#include <iostream>
#include <vector>
#pragma GCC target("avx2")
#pragma GCC optimize("O3")
#pragma GCC optimize("unroll-loops")
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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0