背景:
发现自己看不懂原来的代码了。
题目传送门:
https://www.luogu.org/problemnew/show/P4551
题意:
一棵树,求最长异或路径。
思路:
back to summary
\text{back to summary}
back to summary.
01
Trie
01\text{Trie}
01Trie维护序列异或的模板题。
搞一个从根到当前点的异或前缀
p
r
e
i
pre_i
prei。
因为
x
xor
x
=
0
x \text{ xor }x=0
x xor x=0,所以我们要找到一个点对
(
i
,
j
)
(i,j)
(i,j)使得
p
r
e
i
xor
p
r
e
j
pre_i\text{ xor }pre_j
prei xor prej最大。
对于每一个
p
r
e
i
pre_i
prei,怎么快速找到一个点使得其最大。
贪心地想,每一位互不影响,因此我们只需要尽可能使最高位最大即可。
那就构建一个
01
Trie
01\text{Trie}
01Trie,在里面从上往下找即可。
代码:
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long
using namespace std;
int n,len=0;
LL ans=0;
struct node2{int x,y,z,next;} a[200010];
int last[100010];
LL pre[100010];
struct node1
{
int son[2];
node1(){son[0]=son[1]=0;}
} tr[1000010];
void ins(int x,int y,int z)
{
a[++len]=(node2){x,y,z,last[x]};last[x]=len;
}
void insert(LL x)
{
int root=0;
for(int i=31;i>=0;i--)
{
bool tmp=x&(1LL<<i);
if(!tr[root].son[tmp]) tr[root].son[tmp]=++len;
root=tr[root].son[tmp];
}
}
LL findmax(LL x)
{
int root=0;
LL sum=0;
for(int i=31;i>=0;i--)
{
bool tmp=x&(1<<i);
if(tr[root].son[tmp^1]) root=tr[root].son[tmp^1],sum|=(1<<i); else root=tr[root].son[tmp];
}
return sum;
}
void dfs(int x,int fa)
{
for(int i=last[x];i;i=a[i].next)
{
int y=a[i].y;
if(y==fa) continue;
pre[y]=pre[x]^(LL)a[i].z;
dfs(y,x);
}
}
int main()
{
int x,y,z;
scanf("%d",&n);
for(int i=1;i<n;i++)
{
scanf("%d %d %d",&x,&y,&z);
ins(x,y,z),ins(y,x,z);
}
dfs(1,0);
len=0;
for(int i=1;i<=n;i++)
insert(pre[i]);
for(int i=1;i<=n;i++)
ans=max(ans,findmax(pre[i]));
printf("%lld",ans);
}