HDU 4757 Tree —— 树形可持久化字典树求链上找一个值与x异或最大

This way

题意：

给你一棵树，每个点都有一个权值。有q个询问，每次问你从x到y的路上选一个值与x异或最大是多少。

题解：

从父亲到儿子的顺序建立字典树，每次查询的时候找到lca的父亲l，查询x-l与y-l的最大值即可。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+5;
int tree[N*40][2],son[N*40][2],rt[N],cnt,fa[N][25],dep[N];
int head[N],tot,now;
int a[N];
int n,q;
struct node
{
    int to,next;
}e[N*2];
void add(int x,int y)
{
    e[tot].to=y;
    e[tot].next=head[x];
    head[x]=tot++;
}
void build(int last,int root,int num,int ret)
{
    if(ret==0)
        return ;
    ll a=ret&num;
    int v=(a==ret)?1:0;
    tree[root][v]=tree[last][v]+1;
    tree[root][v^1]=tree[last][v^1];
    son[root][v^1]=son[last][v^1];
    build(son[last][v],son[root][v]=++cnt,num,ret/2);
}

void dfs(int x,int f)
{
    fa[x][0]=f;
    dep[x]=dep[f]+1;
    build(rt[f],rt[x]=++cnt,a[x],(1<<16));
    for(int i=head[x];~i;i=e[i].next)
    {
        int ne=e[i].to;
        if(ne==f)
            continue;
        dfs(ne,x);
    }
}

void deal(){
    for(int j=1;(1<<j)<=n;j++)
        for(int i=1;i<=n;i++)
            fa[i][j]=fa[fa[i][j-1]][j-1];

}

int lca(int x,int y){
    if(dep[x]<dep[y])swap(x,y);
    for(int i=20;i>=0;i--)
        if(dep[fa[x][i]]>=dep[y])
            x=fa[x][i];
    if(x==y)return x;
    for(int i=20;i>=0;i--)if(fa[x][i]!=fa[y][i])x=fa[x][i],y=fa[y][i];
    return fa[x][0];
}
int query(int last,int root,int num,int sum,int ret)
{
    if(ret==0)
        return sum;
    ll a=ret&num;
    int v=(a==ret)?0:1;
    if(tree[root][v]-tree[last][v]>0)
        return query(son[last][v],son[root][v],num,sum+ret,ret/2);
    else
        return query(son[last][v^1],son[root][v^1],num,sum,ret/2);
}
int main()
{

    while(scanf("%d%d",&n,&q)!=EOF)
    {
        memset(head,-1,sizeof(head));
        memset(rt,0,sizeof(rt));
        memset(tree,0,sizeof(tree));
        cnt=tot=now=0;
        for(int i=1;i<=n;i++)
            scanf("%d",&a[i]);
        int x,y;
        ll val;
        for(int i=2;i<=n;i++)
            scanf("%d%d",&x,&y),add(x,y),add(y,x);
        //for(int i=head[1];~i;i=e[i].next)
            //printf("%d\n",e[i].to);
        dfs(1,0);
        deal();

        for(int i=1;i<=q;i++)
        {
            scanf("%d%d%lld",&x,&y,&val);
            int l=fa[lca(x,y)][0];
            int ans=query(rt[l],rt[x],val,0,(1<<16));
            ans=max(ans,query(rt[l],rt[y],val,0,(1<<16)));
            printf("%d\n",ans);
        }
    }
    return 0;
}
/*

8 100
1 2 3 4 3 2 4 2
1 2
1 3
2 4
2 5
5 7
5 8
3 6
2 6 3
*/