【bzoj4103】異或運算(貪心+可持久化Trie)

題意:給定長度爲n 的數列X=x1,x2,...,xn 和長度爲m 的數列Y=y1,y2,...,ym ,令矩陣A 中第i 行第j 列的值Ai,j=xi xor yj ,每次詢問給定矩形區域i[u,d],j[l,r] ,找出第k 大的Ai,j 。數據範圍:0Xi,Yj<231,1udn1000,1lrm300000,1k(du+1)(rl+1),1p500

異或一個數不難想到Trie。
又是區間的的查詢,所以需要持久化。
我們對數列Y 建可持久化Trie。
我們又發現np 都很小,所以對於每一組詢問,我們可以暴力從u 枚舉到d 來處理。
那麼怎麼處理呢?我們可以從高位往低位枚舉,類似名次樹一樣地求,只是這裏是在可持久化Trie上進行處理罷了。
如果有誤在評論區吼一聲哦!
具體的話請看代碼:

#include<cstdio>
#include<algorithm>
using namespace std;
struct Trie{
    int nxt[2],siz;
    Trie(){
        nxt[0]=nxt[1]=siz=0;
    }
}tr[10000010];
int n,m,p,u,d,l,r,k,cnt,x[1010],rt[300010],nowl[1010],nowr[1010];
int rd(){
    int x=0;
    char c;
    do c=getchar();
    while(!isdigit(c));
    do{
        x=(x<<1)+(x<<3)+(c^48);
        c=getchar();
    }while(isdigit(c));
    return x;
}
void insert(int p,int &o,int val,int dep){
    o=++cnt;
    tr[o]=tr[p];
    tr[o].siz++;
    if(dep==-1)
        return;
    int d=(val>>dep)&1;
    insert(tr[p].nxt[d],tr[o].nxt[d],val,dep-1);
    return;
}
int solve(){
    int ret=0;
    for(int i=u;i<=d;i++){
        nowl[i]=rt[l-1];
        nowr[i]=rt[r];
    }
    for(int i=30;~i;i--){
        int tot=0;
        for(int j=u;j<=d;j++)
            tot+=tr[tr[nowr[j]].nxt[((x[j]>>i)&1)^1]].siz-tr[tr[nowl[j]].nxt[((x[j]>>i)&1)^1]].siz;
        if(tot>=k){
            ret+=(1<<i);
            for(int j=u;j<=d;j++){
                nowr[j]=tr[nowr[j]].nxt[((x[j]>>i)&1)^1];
                nowl[j]=tr[nowl[j]].nxt[((x[j]>>i)&1)^1];
            }
        }
        else{
            k-=tot;
            for(int j=u;j<=d;j++){
                nowr[j]=tr[nowr[j]].nxt[(x[j]>>i)&1];
                nowl[j]=tr[nowl[j]].nxt[(x[j]>>i)&1];
            }
        }
    }
    return ret;
}
int main(){
    n=rd();
    m=rd();
    for(int i=1;i<=n;i++)
        x[i]=rd();
    for(int i=1;i<=m;i++){
        int y=rd();
        insert(rt[i-1],rt[i],y,30);
    }
    p=rd();
    while(p--){
        u=rd();
        d=rd();
        l=rd();
        r=rd();
        k=rd();
        printf("%d\n",solve());
    }
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章