可持久化01trie字典树

可持久化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;
}


  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

繁水682

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值