”今日头条杯“湖北省赛 I. Five Day Couple 可持久化01字典树

题目链接

https://www.nowcoder.com/acm/contest/104/H

题意

不带修改区间询问，每次给定一x，l，r。问x与a[l~r]中哪个数异或最大，输出这个最大值。

思路

对每一个前缀a[1~i] 建立一颗树，根结点为rt[i]。树中包含这个区间的所有数，结点中额外加个域cnt，记录有多少个数走到过这个节点。对于查询l,r。我们同时查询树rt[l-1]与树rt[r]。两颗树中的结点的cnt值不同，说明这个数是出现在区间[l,r]的数，接下来的做法就是贪心了，每次往反方向走。
至于每个前缀建树这样是会超内存的，相邻的两个前缀只相差一个数，所以只需修改多出来的logn个结点。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
#define ms(s) memset(s, 0, sizeof(s))
const int INF = 0x3f3f3f3f;
const int maxn = 1e5+10;
#define local
struct trie
{
    int cnt;
    trie *next[2];
    trie()
    {
        for(int i=0; i<2; i++)
            this->next[i]=NULL;
        this->cnt = 1;
    }
    trie(trie *p)
    {
        for(int i=0; i<2; i++)
            this->next[i]=p->next[i];
        this->cnt = p->cnt+1;
    }
};
trie *rt[maxn];
void insert(trie *rt,int x)
{
    trie *p=rt;
    for(int i=30; i>=0; i--)
    {
        int bit = (x>>i)&1;
        if(p->next[bit])p->next[bit]=new trie(p->next[bit]);
        else p->next[bit]=new trie();
        p=p->next[bit];
    }
}
int find(trie *l,trie *r,int x)
{
    int ret=0;
    trie *p1 = l,*p2 = r;
    int lcnt=0,rcnt=0;
    for(int i=30; i>=0; i--)
    {
        int bit = (x>>i)&1;
        lcnt=rcnt=0;
        if(p2&&p2->next[!bit]!=NULL)rcnt=p2->next[!bit]->cnt;
        if(p1&&p1->next[!bit]!=NULL)lcnt=p1->next[!bit]->cnt;
        if(rcnt>lcnt)
        {
            if(p1)p1 = p1->next[!bit];
            if(p2)p2 = p2->next[!bit];
            ret += (1<<i);
        }
        else
        {
            if(p1)p1 = p1->next[bit];
            if(p2)p2 = p2->next[bit];
        }
    }
    return ret;
}
int main()
{
    rt[0] = new trie();
    int n;
    scanf("%d",&n);
    for(int i=1; i<=n; i++)
    {
        int tmp;
        scanf("%d",&tmp);
        rt[i]=new trie(rt[i-1]);
        insert(rt[i],tmp);
    }
    int q;
    scanf("%d",&q);
    while(q--)
    {
        int x,l,r;
        scanf("%d%d%d",&x,&l,&r);
        printf("%d\n",find(rt[l-1],rt[r],x));
    }
}