結果

問題 No.2977 Kth Xor Pair
ユーザー t98slidert98slider
提出日時 2024-12-01 03:43:53
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,881 bytes
コンパイル時間 2,408 ms
コンパイル使用メモリ 213,400 KB
実行使用メモリ 33,040 KB
最終ジャッジ日時 2024-12-01 03:44:00
合計ジャッジ時間 6,686 ms
ジャッジサーバーID
（参考情報） 		judge4 / 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 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
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 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 WA -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
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 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;
    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;
    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()){
            int cnt = 0;
            for(auto &&v : a){
                cnt += (v >> i & 1);
            }
            tot -= cnt * (ll)(n - cnt);
            if(k >= tot){
                k -= tot;
                ans |= 1 << i;
                A.emplace_back(vector<int>({}));
                B.emplace_back(vector<int>({}));
                A[0].reserve(cnt);
                B[0].reserve(n - cnt);
                C.emplace_back(0, 0);
                for(auto &&v : a){
                    if(v >> i & 1) A.back().emplace_back(v);
                    else B.back().emplace_back(v);
                }
            }
            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