“今日头条杯”首届湖北省大学程序设计竞赛 - I - Five Day Couple （可持续化字典树）

题目链接

题意：

有一个有n个数的数列a[]，从1到n标号，有m次询问，询问包含三个数 b,l,r

求 a[l] , ... , a[r] 区间内的一个数与 b 异或的最大值

求异或最大，假如区间均为[1,n]，则可以用字典树，然后每次询问都只用O(1)的效率就可以得到了

就是化为二进制数，然后从位数高的向下推，尽量找不同的

当区间会变时，要判断含有在位置上有0或1的数是否在这个区间里

所以利用可持续化的思想（比如主席树）

保留每个时间点的状态，然后R - (L-1)的字典树就是 L到R区间的字典树

然后就搜一遍就可以了

#include <iostream>
#include <string.h>
#include <stdio.h>
using namespace std;
const int maxn = 1e5 + 10;
int head[maxn],son[maxn*31][2],num[maxn*31][2]; // 一个数为30位，再加上一位头节点，最多为 31 位，所以乘上31
int n,m,x,a[35],l,r,sz = 0;
void add(int now,int be,int pos){ // now 表示当前的 ，be 表示上次的
    if(pos>30) return;
    int val = a[pos];
    son[now][val] = ++sz;
    num[now][val] = num[be][val] + 1; // 当前的个数 = 之前的个数 + 1
    if(num[be][val^1]>0){ //在之前另一边有子树，加入当前树
        son[now][val^1] = son[be][val^1];
        num[now][val^1] = num[be][val^1];
    }
    add(sz, son[be][val], pos+1);
}

void query(int L,int R,int pos){
    if(pos>30) return;
    int val = a[pos]^1; // 将01互换
    if(num[R][val]-num[L][val]>0){ // 说明在所给区间里有和该位相反的数
        //判断下一位
        query(son[L][val], son[R][val], pos+1);
        a[pos] = 1; //所以取这个数，它们异或的结果为1
    }else{
        // 区间内不存在与该位相反的数
        query(son[L][a[pos]], son[R][a[pos]], pos+1);
        a[pos] = 0;
    }
}
void get(){
    for(int j=30;j>=1;j--){
        a[j] = x % 2; x /= 2;
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++) {
        scanf("%d",&x);
        get(); // 得到x的二进制的每一位数 存入数组中
        head[i] = ++sz; // 包含前i个数的字典树的头节点
        add(sz,head[i-1],1); // 建立包含前i个数的字典树
    }
    scanf("%d",&m);
    for(int i=0;i<m;i++){
        scanf("%d%d%d",&x,&l,&r);
        get();
        query(head[l-1],head[r],1);
        x = 0;
        for(int j=1;j<=30;j++) x = x * 2 + a[j];
        printf("%d\n",x);
    }
    return 0;
}