可持久化01trie字典树是个比较复杂的数据结构,和主席树的构成原理很相似。区别就在主席树构造时候左右子树都是显示的,而可持久化01的左右(0或1)其中的一个是隐藏的,一开始看会有些抽象。
首先,什么是可持久化?
在我们解决一些问题的时候,我们想要建立多个数据结构来维护l~r的区间:
以线段树为例
我们建造1~1的线段树,1~2状态的线段树,1~3状态的线段树,1~4状态的线段树。。。
但这样建造,复杂度太高了,所以我们需要优化空间。把这样的多颗线段树建在一起:
这样大大节省了空间。
同理,可持续化01trie也是这样。
封装后的板子如下:
int n,m;
int root[N],ver[N];
int idx=0;
int s[N];//01串
struct long_trie{
int tr[N*33][2],ver[N*33];
void insert(int x,int y,int i){
ver[x]=i;
for (int k=31;k>=0;k--){
int c=(s[i]>>k)&1;
tr[x][!c]=tr[y][!c];//复制前面节点
tr[x][c]=++idx;//新开节点
x=tr[x][c];y=tr[y][c];
ver[x]=i;
}
}
int query(int l,int r,int v){
int x=root[r],ans=0;
for (int k=31;k>=0;k--){
int c=(v>>k)&1;
//如果版本小于l,或空节点就只能和v的当前位置一样了
if (ver[tr[x][!c]]>=l){
x=tr[x][!c],ans+=1<<k;
}
else x=tr[x][c];
}
}
}tr;
例题:
XOR on Tree
题目意思:给一棵树,树上每个节点有权值,给q个询问,每次询问u,v。让你在v的所有子树中(包括v)找到权值和u的异或最大。
我们可以用dfn序把树上问题转化成区间问题,然后就可以直接套可持久化的板子了。
#include <bits/stdc++.h>
#define endl '\n'
#define int long long
//#define double long double
#define pii pair<int,int>
#define pb push_back
#define rep(i,l,r) for (int i=l;i<=r;i++)
#define nep(i,r,l) for (int i=r;i>=l;i--)
#define CIO std::ios::sync_with_stdio(false);
#define cio cin.tie(0);cout.tie(0);
#define pc putchar
using namespace std;
const int INF=0x3f3f3f3f;
const int LINF=0x3f3f3f3f3f3f3f3f;
const double eps=1e-6;
const int N=2e5+5;
const double pi=acos(-1.0);
const int mod=1e9+7;
int fp(int a,int x){int ans=1;while (x){if (x&1) ans=ans*a;a=a*a;x>>=1;}return ans;}
int fpm(int a,int x){int ans=1;while (x){if (x&1) ans=ans*a%mod;a=a*a%mod;x>>=1;}return ans;}
inline void read(int &x){
x=0;short flag=1;char c=getchar();
while(c<'0'||c>'9'){if(c=='-')flag=-1;c=getchar();}
while(c>='0'&&c<='9'){x=(x<<3)+(x<<1)+(c^48);c=getchar();}
x*=flag;
}
inline void write(int x){
if(x<0){putchar('-');x=-x;}if(x>9) write(x/10);
putchar(x%10+'0');
}
vector<int> ve[N];
int dfn[N],bol[N],siz[N],tim=0;
void dfs(int u,int fa){
dfn[u]=++tim;
bol[tim]=u;
siz[u]=1;
for (int i=0;i<ve[u].size();i++){
int v=ve[u][i];
if (v==fa) continue;
dfs(v,u);
siz[u]+=siz[v];
}
}
int root[N],ver[N];
int s[N],idx=0;
struct long_trie{
int tr[N*33][2],ver[N*33];
void insert(int x,int y,int i){
ver[x]=i;
for (int k=31;k>=0;k--){
int c=(s[bol[i]]>>k)&1;
tr[x][!c]=tr[y][!c];
tr[x][c]=++idx;
x=tr[x][c];y=tr[y][c];
ver[x]=i;
}
}
int query(int l,int r,int w){
int x=root[r],ans=0;
for (int k=31;k>=0;k--){
int c=(w>>k)&1;
if (ver[tr[x][!c]]>=l){
x=tr[x][!c],ans+=(1<<k);
}
else x=tr[x][c];
}
return ans;
}
}trie;
void work(){
int n;cin>>n;
for (int i=1;i<=n;i++){
cin>>s[i];
}
for (int i=1;i<=n-1;i++){
int u,v;cin>>u>>v;
ve[u].pb(v);
ve[v].pb(u);
}
dfs(1,0);
ver[0]=-1;
for (int i=1;i<=n;i++){
root[i]=++idx;
trie.insert(root[i],root[i-1],i);
}
int q;cin>>q;
for (int i=1;i<=q;i++){
int u,v;cin>>u>>v;
cout<<trie.query(dfn[v],dfn[v]+siz[v]-1,s[u])<<endl;
}
}
signed main(){
CIO;
cio;
//多组样例别忘了清空数组,stl
//int _;cin>>_;while(_--)
{
work();
}
return 0;
}