結果

問題 No.2977 Kth Xor Pair
ユーザー t98slidert98slider
提出日時 2024-12-01 04:34:34
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 3,805 bytes
コンパイル時間 2,650 ms
コンパイル使用メモリ 216,316 KB
実行使用メモリ 84,044 KB
最終ジャッジ日時 2024-12-01 04:34:47
合計ジャッジ時間 8,828 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using ll = long long;

int main(){
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n;
    ll k, tot = 0;
    cin >> n >> k;
    vector<pair<int,int>> C;
    vector<vector<int>> A, B, D(1);
    C.reserve(n);
    A.reserve(n);
    B.reserve(n);
    tot = (ll)(n) * (n - 1) / 2;
    vector<int> a(n);
    for(auto &&v : a) cin >> v;
    D[0] = a;
    auto f = [&](vector<int> &a, vector<int> &b, int lg){
        int cnt0 = 0, cnt1 = 0;
        for(auto &&v : a) cnt0 += (v >> lg & 1);
        for(auto &&v : a) cnt1 += (v >> lg & 1);
        return make_pair(cnt0, cnt1);
    };
    int ans = 0;
    for(int i = __lg(*max_element(a.begin(), a.end())), prv = 0; i >= 0; i--){
        if(A.empty()){
            for(auto &&vec : D){
                int cnt = 0;
                for(auto &&v : vec){
                    cnt += (v >> i & 1);
                }
                tot -= cnt * (ll)(vec.size() - cnt);
            }
            if(k >= tot){
                k -= tot;
                ans |= 1 << i;
                for(auto &&vec : D){
                    A.emplace_back(vector<int>({}));
                    B.emplace_back(vector<int>({}));
                    C.emplace_back(0, 0);
                    for(auto &&v : a){
                        if(v >> i & 1) A.back().emplace_back(v);
                        else B.back().emplace_back(v);
                    }
                }
            }else{
                int sz = D.size();
                for(int j = 0; j < sz; j++){
                    auto &&vec = D[j];
                    int cnt = 0;
                    for(auto &&v : vec){
                        if(v >> i & 1) cnt++;
                    }
                    if(cnt != 0 && cnt != vec.size()){
                        D.emplace_back(vector<int>());
                        D.back().reserve(cnt);
                        for(int k = vec.size() - 1; k >= 0; k--){
                            if(vec[k] >> i & 1){
                                D.back().emplace_back(vec[k]);
                                swap(vec[k], vec.back());
                                vec.pop_back();
                            }
                        }
                    }
                }
            }
            continue;
        }
        ll sv = 0;
        for(int j = prv; j < A.size(); j++){
            C[j] = f(A[j], B[j], i);
            auto [a1, b1] = C[j];
            ll a0 = A[j].size() - a1, b0 = B[j].size() - b1;
            sv += a1 * b0 + b1 * a0;
        }
        tot -= sv;
        if(k >= tot){
            k -= tot;
            ans |= 1 << i;
            int sz = A.size();
            for(int j = prv; j < sz; j++){
                auto [a1, b1] = C[j];
                int a0 = A[j].size() - a1, b0 = B[j].size() - b1;
                if(a0 >= 1 && b1 >= 1){
                    A.emplace_back(vector<int>({}));
                    B.emplace_back(vector<int>({}));
                    C.emplace_back(0, 0);
                    A.back().reserve(a0);
                    B.back().reserve(b1);
                    for(auto &&v : A[j]) if(~v >> i & 1) A.back().emplace_back(v);
                    for(auto &&v : B[j]) if(v >> i & 1) B.back().emplace_back(v);
                }
                if(a1 >= 1 && b0 >= 1){
                    A.emplace_back(vector<int>({}));
                    B.emplace_back(vector<int>({}));
                    C.emplace_back(0, 0);
                    A.back().reserve(a1);
                    B.back().reserve(b0);
                    for(auto &&v : A[j]) if(v >> i & 1) A.back().emplace_back(v);
                    for(auto &&v : B[j]) if(~v >> i & 1) B.back().emplace_back(v);
                }
            }
            prv = sz;
        }
    }
    cout << ans << '\n';
}
0