HDU4757：Tree——题解

http://acm.hdu.edu.cn/showproblem.php?pid=4757

给一棵有点值的树，每次询问u~v的最短路当中的一个点的点权异或z最大值。

前置技能：HDU4825

前置技能：BZOJ3261（可持久化trie）

默认你会如何建可持久化trie跑贪心了。

emmmm既然你已经懂了那还说啥，u~v的链可以剖成u~lca和v~lca，即每次询问这两个链和z异或的最大值即可。

PS：注意询问的时候区间左端点-1，即询问的是rt[fa(lca)]~rt[u/v]。

#include<cstdio>
#include<iostream>
#include<queue>
#include<cstring>
#include<algorithm>
#include<cctype>
using namespace std;
const int N=1e5+;
inline int read(){
    int X=,w=;char ch=;
    while(!isdigit(ch)){w|=ch=='-';ch=getchar();}
    while(isdigit(ch))X=(X<<)+(X<<)+(ch^),ch=getchar();
    return w?-X:X;
}
struct trie{
    int son[],sum;
}tr[*N];
struct node{
    int to,nxt;
}e[N*];
int tot,a[N],rt[N],pool,cnt,head[N];
int dep[N],anc[N][];
inline void add(int u,int v){
    e[++cnt].to=v;e[cnt].nxt=head[u];head[u]=cnt;
}
inline void addtwo(int u,int v){
    add(u,v);add(v,u);
}
void insert(int y,int &x,int k,int now){
    tr[x=++pool]=tr[y];
    tr[x].sum++;
    if(now<)return;
    bool p=k&(<<now);
    insert(tr[y].son[p],tr[x].son[p],k,now-);
    return;
}
int query(int nl,int nr,int k,int now){
    if(now<)return ;
    bool p=k&(<<now);
    int delta=tr[tr[nr].son[p^]].sum-tr[tr[nl].son[p^]].sum;
    if(delta>)return (<<now)+query(tr[nl].son[p^],tr[nr].son[p^],k,now-);
    else return query(tr[nl].son[p],tr[nr].son[p],k,now-);
}
void dfs(int u,int f){
    anc[u][]=f;
    for(int k=;k<=;k++)
    anc[u][k]=anc[anc[u][k-]][k-];
    dep[u]=dep[f]+;
    insert(rt[f],rt[u],a[u],);
    for(int i=head[u];i;i=e[i].nxt){
        int v=e[i].to;
        if(v!=f)dfs(v,u);
    }
    return;
}
inline int LCA(int i,int j){
    if(dep[i]<dep[j])swap(i,j);
    for(int k=;k>=;k--){
        if(dep[anc[i][k]]>=dep[j])i=anc[i][k];
    }
    if(i==j)return i;
    for(int k=;k>=;k--){
        if(anc[i][k]!=anc[j][k])i=anc[i][k],j=anc[j][k];
    }
    return anc[i][];
}
inline void init(){
    memset(tr,,sizeof(tr));
    memset(rt,,sizeof(rt));
    memset(head,,sizeof(head));
    tot=;cnt=pool=;
}
int main(){
    int n,m;
    while(scanf("%d%d",&n,&m)!=EOF){
    init();
    for(int i=;i<=n;i++)a[i]=read();
    for(int i=;i<n;i++)addtwo(read(),read());
    dfs(,);
    for(int i=;i<=m;i++){
        int u=read(),v=read(),x=read(),lca=LCA(u,v),fa=anc[lca][];
        printf("%d\n",max(query(rt[fa],rt[u],x,),
                  query(rt[fa],rt[v],x,)));
    }
    }
    return ;
}

+++++++++++++++++++++++++++++++++++++++++++

+本文作者：luyouqi233。 　　　　　　　　　　　　　　+

+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/+

+++++++++++++++++++++++++++++++++++++++++++