結果

問題 No.2977 Kth Xor Pair
ユーザー The Forsaking
提出日時 2024-12-25 23:38:05
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,817 ms / 3,000 ms
コード長 3,404 bytes
コンパイル時間 1,230 ms
コンパイル使用メモリ 114,956 KB
最終ジャッジ日時 2025-02-26 16:50:45
ジャッジサーバーID
（参考情報） 		judge3 / judge5
純コード判定しない問題か言語
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 34
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:110:42: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
  110 |     for (int i = 1; i < n + 1; i++) scanf("%d", w + i), insert(w[i]);
      |                                     ~~~~~^~~~~~~~~~~~~

ソースコード

diff # 

#include <iostream>
#include <sstream>
#include <iomanip>
#include <cstring>
#include <string>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <vector>
#include <queue> 
#include <unordered_set>
#include <unordered_map>
#include <bitset>
#include <ctime>
#include <assert.h>
#include <deque>
#include <list>
#include <stack>


using namespace std;

#define is_mul_overflow(a, b) \
    ((b != 0) && (a > LLONG_MAX / b || a < LLONG_MIN / b))
 
typedef pair<long long, int> pli;
typedef pair<int, long long> pil;
typedef pair<long long , long long> pll;
typedef pair<int, int> pii;
typedef pair<double, double> pdd;
typedef pair<int, pii> piii;
typedef pair<int, long long > pil;
typedef pair<long long, pii> plii;
typedef pair<double, int> pdi;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ull, ull> puu;
typedef long double ld;
const int N = 2000086, MOD = 998244353, INF = 0x3f3f3f3f, MID = 333;
const long double EPS = 1e-8;
int dx[4] = {1, 0, -1, 0}, dy[4] = {0, 1, 0, -1};
// int dx[8] = {1, 1, 0, -1, -1, -1, 0, 1}, dy[8] = {0, 1, 1, 1, 0, -1, -1, -1};
// int dx[8] = {2, 1, -1, -2, -2, -1, 1, 2}, dy[8] = {1, 2, 2, 1, -1, -2, -2, -1};
int n, m, cnt;
int w[N];
vector<ll> num;
ll res;

ll lowbit(ll x) { return x & -x; }
ll gcd(ll a, ll b) { return b ? gcd(b, a % b) : a; }
ll lcm(ll a, ll b) { return a / gcd(a, b) * b; }
inline double rand(double l, double r) { return (double)rand() / RAND_MAX * (r - l) + l; }
inline ll qmi(ll a, ll b, ll c) { ll res = 1; while (b) { if (b & 1) res = res * a % c; a = a * a % c; b >>= 1; } return res; }
inline ll qmi(ll a, ll b) { ll res = 1; while (b) { if (b & 1) res *= a; a *= a; b >>= 1; } return res; }
inline double qmi(double a, ll b) { double res = 1; while (b) { if (b & 1) res *= a; a *= a; b >>= 1; } return res; } 
// inline ll C(ll a, ll b) { if (a < b) return 0; if (b > a - b) b = a - b; ll res = 1; for (ll i = 1, j = a; i <= b; i++, j--) { res = res * (j % MOD) % MOD; res = res * qmi(i, MOD - 2, MOD) % MOD; } return res; }
inline ll C(ll a, ll b, int* c) { if (a < b) return 0; ll res = 1; for (ll j = a, i = 1; i < b + 1; i++, j--) res *= j; for (ll j = a, i = 1; i < b + 1; i++, j--) res /= i; return res; }
inline int find_(int x) { return lower_bound(num.begin(), num.end(), x) - num.begin(); }

ll k;
int tr[N * 10][2], c[N * 10];

void insert(int x) {
    int p = 0;
    for (int i = 30; i > -1; i--) {
        int v = x >> i & 1;
        if (!tr[p][v]) tr[p][v] = ++cnt;
        p = tr[p][v];
        c[p]++;
    }
}

bool check(int mid) {
    ll res = 0;
    res -= n;

    for (int i = 1; i < n + 1; i++) {
        int p = 0, j = 30;
        for (; j > -1; j--) {
            int v = mid >> j & 1, t = w[i] >> j & 1;
            if (v) {
                res += c[tr[p][t]];
                if (!tr[p][t ^ 1]) break;
                p = tr[p][t ^ 1];
            } else {
                if (!tr[p][t]) break;
                p = tr[p][t];
            }
        }
        if (j == -1) res += c[p];
    }
    res >>= 1;
    return res < k;
}

inline void solve() {
    ll l = 0, r = 2e9;
    while (l < r) {
        int mid = l + r >> 1;
        // c(mid) < K
        if (check(mid)) l = mid + 1;
        else r = mid;
    }
    printf("%lld\n", l);
}

int main() {
    cin >> n >> k;
    for (int i = 1; i < n + 1; i++) scanf("%d", w + i), insert(w[i]);
    solve();
    return 0;
}
0