hdu 4757 Tree(可持久化Trie)

最新推荐文章于 2021-10-24 23:37:39 发布

原创最新推荐文章于 2021-10-24 23:37:39 发布 · 1.6k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#data struct

数据结构专栏收录该内容

91 篇文章

订阅专栏

该博客介绍了如何使用可持久化Trie解决一类树上路径权值异或最大值的查询问题。通过将节点权值转换为二进制并插入Trie，利用可持久化特性来查询历史信息，对于每个询问，从起点到终点在Trie中搜索，根据异或操作更新结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题意：给出一棵树，每个点上有权值，有m组询问，每次询问从x到y的路径上的权值与z异或的最大值为多少。

思路：把每个点的权值用二进制表示，然后插入Trie。利用可持久化Trie可以查询历史信息的特点，每个节点保存从当前点到根节点的Trie。对于每个查询，在Trie中查询，对于每一位，寻找与z对应位相反的分支，如果有，结果res|=(1<<i)，继续查询，否则，查找与z对应位相同的分支。

代码：

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<stack>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
#define Inf 0x3FFFFFFFFFFFFFFFLL
#define eps 1e-9
#define pi acos(-1.0)
using namespace std;
typedef long long ll;
const int maxn=100000+10;
const int maxm=3000000+10;
const int lev=15;
int ch[maxm][2],sum[maxm],size;
int root[maxn];
int Newnode(int val)
{
   size++;
   ch[size][0]=ch[size][1]=0;
   sum[size]=val;
   return size;
}
int Insert(int x,int val,int dep)
{
    int rt=Newnode(sum[x]+1);
    if(dep==-1) return rt;
    int d=(val>>dep)&1;
    ch[rt][d^1]=ch[x][d^1];
    ch[rt][d]=Insert(ch[x][d],val,dep-1);
    return rt;
}
int query(int x,int y,int val)
{
    int res=0;
    for(int i=lev;i>=0;--i)
    {
        int d=(val>>i)&1;
        int tmp=sum[ch[y][d^1]]-sum[ch[x][d^1]];
        if(tmp>0) res|=(1<<i);
        else d^=1;
        x=ch[x][d^1];y=ch[y][d^1];
    }
    return res;
}
struct Edge
{
    int v,next;
    Edge(){};
    Edge(int vv,int nx){v=vv;next=nx;}
}edges[maxn<<1];
int head[maxn],pa[maxn],father[maxn],nEdge,n,m;
vector<pair<int,int> >querys[maxn];
int ans[maxn],w[maxn],z[maxn];
bool vis[maxn];
void Init()
{
    ch[0][0]=ch[0][1]=sum[0]=size=0;
    memset(root,0,sizeof(root));
    memset(head,0xff,sizeof(head));
    memset(vis,0,sizeof(vis));
    nEdge=-1;father[1]=0;
    for(int i=0;i<=n;++i) querys[i].clear();
}
int Find(int x)
{
    return x==pa[x]?x:pa[x]=Find(pa[x]);
}
void AddEdges(int u,int v)
{
    edges[++nEdge]=Edge(v,head[u]);
    head[u]=nEdge;
    edges[++nEdge]=Edge(u,head[v]);
    head[v]=nEdge;
}
void dfs(int u,int fa)
{
    pa[u]=u;vis[u]=true;
    root[u]=Insert(root[father[u]],w[u],lev);
    int sz=querys[u].size();
    int v,pos,lca;
    for(int i=0;i<sz;++i)
    {
        v=querys[u][i].first;pos=querys[u][i].second;
        if(vis[v])
        {
            lca=Find(v);
            ans[pos]=max(query(root[lca],root[u],z[pos]),query(root[lca],root[v],z[pos]));
            ans[pos]=max(w[lca]^z[pos],ans[pos]);
        }
    }
    for(int k=head[u];k!=-1;k=edges[k].next)
    {
        v=edges[k].v;
        if(v==fa) continue;
        father[v]=u;
        dfs(v,u);
        pa[v]=u;
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    while(~scanf("%d%d",&n,&m))
    {
        Init();
        for(int i=1;i<=n;++i)
            scanf("%d",&w[i]);
        int u,v;
        for(int i=1;i<n;++i)
        {
            scanf("%d%d",&u,&v);
            AddEdges(u,v);
        }
        for(int i=1;i<=m;++i)
        {
            scanf("%d%d%d",&u,&v,&z[i]);
            querys[u].push_back(make_pair(v,i));
            querys[v].push_back(make_pair(u,i));
        }
        dfs(1,-1);
        for(int i=1;i<=m;++i)
            printf("%d\n",ans[i]);
    }
    return 0;
}