标签:ase ssi 自动机 who swap 根据 algo parameter cte
Time Limit: 10000/5000 MS (Java/Others) Memory Limit: 102400/102400 K (Java/Others)
Total Submission(s): 2058 Accepted Submission(s): 599
#include<cstdio>
#include<cstring>
#include<algorithm>
const int N=100008;
int a[N],head[N],tot,index,cont,n,m;
int root[N],tree[N*35][2],son[N*35][2];
int fa[N],depth[N],up[N][18],pt[N];
struct node{
int next,to;
}e[N<<1];
void add(int u,int v){
e[tot].next=head[u];e[tot].to=v;head[u]=tot++;
e[tot].next=head[v];e[tot].to=u;head[v]=tot++;
}
void build(int last,int cur,int num,int pos){
if(pos<0) return;
int temp=!!(num&(1<<pos));
tree[cur][temp]=tree[last][temp]+1;
son[cur][temp^1]=son[last][temp^1];
tree[cur][temp^1]=tree[last][temp^1];
build(son[last][temp],son[cur][temp]=++cont,num,pos-1);//这里可以看到每一个数字都建立了31个节点,其实只要17个就够了根据题意
}
void dfs(int u){
pt[u]=++index;//记录一下每个节点在树中的位置
build(root[pt[fa[u]]],root[pt[u]]=++cont,a[u],31);
for(int i=head[u];~i;i=e[i].next){
int v=e[i].to;
if(fa[u]==v) continue;
fa[v]=u;
depth[v]=depth[u]+1;
dfs(v);
}
}
void doit(){
for(int i=1;i<=n;++i) up[i][0]=fa[i];
for(int j=1;j<=16;++j) for(int i=1;i<=n;++i) up[i][j]=up[up[i][j-1]][j-1];
}
int lca(int x,int y){
if(depth[x]<depth[y]) std::swap(x,y);
int dt=depth[x]-depth[y];
for(int i=0;i<=16;++i) if(dt&(1<<i)) x=up[x][i];
if(x==y) return x;
for(int i=16;i>=0;--i) if(up[x][i]!=up[y][i]) x=up[x][i],y=up[y][i];
return up[x][0];
}
int query(int last,int cur,int num,int sum,int pos){
if(pos<0) return sum;
int temp=!!(num&(1<<pos));
if(tree[cur][temp^1]-tree[last][temp^1]>0) return query(son[last][temp^1],son[cur][temp^1],num,sum|(1<<pos),pos-1);
else return query(son[last][temp],son[cur][temp],num,sum,pos-1);
}
int main(){
int x,y,z;
while(scanf("%d%d",&n,&m)!=EOF){
memset(head,-1,sizeof(head));
index=tot=cont=0;
for(int i=1;i<=n;++i) scanf("%d",&a[i]);
for(int i=1;i<n;++i) {scanf("%d%d",&x,&y);add(x,y);}
dfs(1);
doit();
while(m--){
scanf("%d%d%d",&x,&y,&z);
int ct=lca(x,y);
int mx=query(root[pt[fa[ct]]],root[pt[x]],z,0,31);
mx=std::max(query(root[pt[fa[ct]]],root[pt[y]],z,0,31),mx);
printf("%d\n",mx);
}
}
}
标签:ase ssi 自动机 who swap 根据 algo parameter cte
原文地址:http://www.cnblogs.com/mfys/p/7466575.html