BZOJ3689 异或之

最新推荐文章于 2020-08-13 22:55:41 发布

原创最新推荐文章于 2020-08-13 22:55:41 发布 · 1.4k 阅读

0 ·

CC 4.0 BY-SA版权

Persistent Data Structures 同时被 2 个专栏收录

3 篇文章

订阅专栏

Heap

3 篇文章

订阅专栏

该博客介绍了如何解决BZOJ3689题目的思路，与NOI2010超级钢琴问题类似。方法是使用全局堆存放每个数与后继数异或的最小值，并在每次出堆时重新插入第k+1小值，重复k次。核心是利用可持久化trie数据结构，在树上二分查找，实现时间复杂度为O(klogn)。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意：给定n个非负整数A[1], A[2], ……, A[n]。
对于每对(i, j)满足1 <= i < j <= n，得到一个新的数A[i] xor A[j]，这样共有n*(n-1)/2个新的数。求这些数（不包含A[i]）中前k小的数。

注：xor对应于pascal中的“xor”，C++中的“^”。

思路：同NOI2010超级钢琴。http://blog.youkuaiyun.com/wyfcyx_forever/article/details/40400327

我们只需一开始在全局堆中放进去每个数和他之后的数异或的最小值，然后在每次出堆的的时候再放进堆中第k+1小值即可。这样重复k次。

那么问题就变成了快速求某段区间给定一个数进行xor的第k小值，我们利用可持久化trie维护size直接在树上二分即可。

时间复杂度依旧O(klogn).

Code:

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
 
#define N 100010
int w[N];
 
int ch[N<<5][2], size[N<<5], ind;
int root[N];
int Newadd(int Last, int bit, int x) {
    int q = ++ind;
    ch[q][0] = ch[Last][0], ch[q][1] = ch[Last][1];
    size[q] = size[Last] + 1;
     
    if (bit < 0)
        return q;
     
    if ((x >> bit) & 1)
        ch[q][1] = Newadd(ch[Last][1], bit - 1, x);
    else
        ch[q][0] = Newadd(ch[Last][0], bit - 1, x);
    return q;
}
int getkth(int Left, int Right, int x, int k) {
    int res = 0, better;
    bool d;
    for(int dep = 30; dep >= 0; --dep) {
        d = (x >> dep) & 1;
        better = size[ch[Right][d]] - size[ch[Left][d]];
        if (better >= k)
            Left = ch[Left][d], Right = ch[Right][d];
        else
            Left = ch[Left][d ^ 1], Right = ch[Right][d ^ 1], res += (1<<dep), k -= better;
    }
     
    return res;
}
 
struct st {
    int end, kth, val;
    st(int _end = 0, int _kth = 0, int _val = 0):end(_end),kth(_kth),val(_val){}
    bool operator < (const st &B) const {
        return val < B.val;
    }
};
 
#define K 250010
struct Heap {
    st a[K];
    int top;
    Heap():top(0){}
    inline void up(int x) {
        for(; x != 1; x >>= 1)
            if (a[x]<a[x>>1])
                swap(a[x],a[x>>1]);
            else
                break;
    }
    inline void down(int x) {
        int son;
        for(; (x<<1)<=top; ) {
            son=(((x<<1)==top)||(a[x<<1]<a[(x<<1)|1]))?(x<<1):((x<<1)|1);
            if (a[son]<a[x])
                swap(a[son],a[x]),x=son;
            else
                break;
        }
    }
    inline void push(const st &x) {
        a[++top] = x;
        up(top);
    }
    st Min() {
        return a[1];
    }
    inline void pop() {
        a[1] = a[top--];
        down(1);
    }
}H;
 
int main() {
    int n, k;
    scanf("%d%d", &n, &k);
     
    register int i, j;
    for(i = 1; i <= n; ++i) {
        scanf("%d", &w[i]);
        root[i] = Newadd(root[i - 1], 30, w[i]);
    }
     
    for(i = 2; i <= n; ++i)
        H.push(st(i, 1, getkth(root[0], root[i - 1], w[i], 1)));
     
    for(i = 1; i <= k; ++i) {
        st tmp = H.Min();
        H.pop();
        if (i != 1)
            putchar(' ');
        printf("%d", tmp.val);
        if (tmp.kth != tmp.end - 1)
            H.push(st(tmp.end, tmp.kth + 1, getkth(root[0], root[tmp.end - 1], w[tmp.end], tmp.kth + 1)));
    }
     
    return 0;
}