P4592 [TJOI2018]异或
题目描述
现在有一颗以\(1\)为根节点的由\(n\)个节点组成的树,树上每个节点上都有一个权值\(v_i\)。现在有\(Q\)次操作,操作如下:
1 x y
:查询节点\(x\)的子树中与\(y\)异或结果的最大值2 x y
:查询路径\(x\)到\(y\)上点与\(z\)异或结果最大值
输入输出格式
输入格式:
第一行是两个数字\(n,Q\);
第二行是\(n\)个数字用空格隔开,第\(i\)个数字\(v_i\)表示点\(i\)上的权值
接下来\(n-1\)行,每行两个数,\(x,y\),表示节点\(x\)与\(y\)之间有边
接下来\(Q\)行,每一行为一个查询,格式如上所述.
输出格式:
对于每一个查询,输出一行,表示满足条件的最大值。
说明
对于\(10\%\)的数据,有\(1<n,Q\leq100\)
对于\(20\%\)的数据,有\(1<n,Q\leq1000\)
对于\(40\%\)的数据,有\(1<n,Q\leq10000\)
对于\(100\%\)的数据,有\(1<n,Q\leq100000\)
对于\(100\%\)的数据,有查询\(1\)中的\(y\leq2^{30}\),查询\(2\)中的\(z\leq2^{30}\)。
区间异或最大值可以用可持久化字典树实现。
这个题建对DFS序建一颗,对自根向下的链建,分别处理两种询问就可以了。
Code:
#include <cstdio>
#include <cctype>
const int N=1e5+10;
int read()
{
int x=0;char c=getchar();
while(!isdigit(c)) c=getchar();
while(isdigit(c)) {x=x*10+c-'0';c=getchar();}
return x;
}
int max(int x,int y){return x>y?x:y;}
#define ls ch[now][0]
#define rs ch[now][1]
#define rep(i,a,b) for(int i=a;i<=b;i++)
int head[N],to[N<<1],Next[N<<1],cnt;
void add(int u,int v)
{
to[++cnt]=v,Next[cnt]=head[u],head[u]=cnt;
}
int poi[N],n,Q;
namespace work1//子树
{
int ch[N*32][2],mx[N*32],tot,dfn[N],low[N],ha[N],dfsclock,root[N];
void dfs(int now,int fa)
{
dfn[now]=++dfsclock;
ha[dfsclock]=now;
for(int i=head[now];i;i=Next[i])
if(to[i]!=fa)
dfs(to[i],now);
low[now]=dfsclock;
}
void Insert(int las,int &now,int dep,int id)
{
if(!now) now=++tot;
if(dep<0){mx[now]=id;return;}
int bit=poi[ha[id]]>>dep&1;
Insert(ch[las][bit],ch[now][bit],dep-1,id);
ch[now][bit^1]=ch[las][bit^1];
mx[now]=max(mx[ls],mx[rs]);
}
void init()
{
dfs(1,0);
rep(i,1,n) Insert(root[i-1],root[i],30,i);
}
int query(int now,int les,int dep,int x)
{
if(dep<0) return x^poi[ha[mx[now]]];
int bit=x>>dep&1;
if(mx[ch[now][bit^1]]>=les) return query(ch[now][bit^1],les,dep-1,x);
return query(ch[now][bit],les,dep-1,x);
}
void work()
{
int x=read(),y=read();
printf("%d\n",query(root[low[x]],dfn[x],30,y));
}
}
namespace work2
{
int f[N][20],ch[N*32][2],mx[N*32],root[N],dep[N],en[N*32],tot,tmp;
void swap(int &x,int &y){tmp=x,x=y,y=tmp;}
void Insert(int las,int &now,int de,int id)
{
if(!now) now=++tot;
if(de<0){mx[now]=dep[id],en[now]=id;return;}
int bit=poi[id]>>de&1;
Insert(ch[las][bit],ch[now][bit],de-1,id);
ch[now][bit^1]=ch[las][bit^1];
mx[now]=max(mx[ls],mx[rs]);
}
void dfs(int now)
{
for(int i=1;f[now][i-1];i++) f[now][i]=f[f[now][i-1]][i-1];
for(int i=head[now];i;i=Next[i])
{
int v=to[i];
if(v==f[now][0]) continue;
dep[v]=dep[now]+1;
f[v][0]=now;
Insert(root[now],root[v],30,v);
dfs(v);
}
}
void init()
{
dep[1]=1;
Insert(0,root[1],30,1);
dfs(1);
}
int LCA(int x,int y)
{
if(dep[x]<dep[y]) swap(x,y);
for(int i=18;~i;i--)
if(dep[f[x][i]]>=dep[y])
x=f[x][i];
if(x==y) return x;
for(int i=18;~i;i--)
if(f[x][i]!=f[y][i])
x=f[x][i],y=f[y][i];
return f[x][0];
}
int query(int now,int les,int de,int x)
{
if(de<0) return poi[en[now]]^x;
int bit=x>>de&1;
if(mx[ch[now][bit^1]]>=les) return query(ch[now][bit^1],les,de-1,x);
return query(ch[now][bit],les,de-1,x);
}
void work()
{
int x=read(),y=read(),z=read(),lca=LCA(x,y);
printf("%d\n",max(query(root[x],dep[lca],30,z),query(root[y],dep[lca],30,z)));
}
}
int main()
{
n=read(),Q=read();
rep(i,1,n) scanf("%d",poi+i);
for(int u,v,i=1;i<n;i++)
u=read(),v=read(),add(u,v),add(v,u);
work1::init();
work2::init();
rep(i,1,Q)
{
if(read()==1) work1::work();
else work2::work();
}
return 0;
}
2018.11.3