码迷,mamicode.com
首页 > 其他好文 > 详细

Luogu P4551 最长异或路径 01trie

时间:2019-06-13 01:06:42      阅读:26      评论:0      收藏:0      [点我收藏+]

标签:etc   fir   pre   for   map   max   ace   不同   不同的   

做一个树上前缀异或和,然后把前缀和插到$01trie$里,然后再对每一个前缀异或和整个查一遍,在树上从高位向低位贪心,按位优先选择不同的,就能贪出最大的答案。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<cctype>
#include<cstdlib>
#include<vector>
#include<queue>
#include<map>
#include<set>
#define ll long long
#define R register int
using namespace std;
namespace Fread {
    static char B[1<<15],*S=B,*D=B;
    #define getchar() (S==D&&(D=(S=B)+fread(B,1,1<<15,stdin),S==D)?EOF:*S++)
    inline int g() {
        R ret=0,fix=1; register char ch; while(!isdigit(ch=getchar())) fix=ch==-?-1:fix;
        do ret=ret*10+(ch^48); while(isdigit(ch=getchar())); return ret*fix;
    }
}using Fread::g;
const int N=100010;
int cnt,n,m,tot;
int vr[N<<1],nxt[N<<1],w[N<<1],fir[N],d[N];
int t[N*32][2];
inline void add(int u,int v,int ww) {vr[++cnt]=v,w[cnt]=ww,nxt[cnt]=fir[u],fir[u]=cnt;}
inline void dfs(int u,int fa) {
    for(R i=fir[u];i;i=nxt[i]) { R v=vr[i];
        if(v==fa) continue; d[v]=d[u]^w[i]; dfs(v,u);
    }
}
inline void ins(int x) { R now=0;
    for(R i=30;~i;--i) {
        R ch=(x>>i)&1; if(!t[now][ch]) t[now][ch]=++tot;
        now=t[now][ch];
    }
}
inline int query(int x) { R now=0,ret=0;
    for(R i=30;~i;--i) {
        R ch=(x>>i)&1; if(!t[now][!ch]) now=t[now][ch];
        else now=t[now][!ch],ret+=(1<<i);
    } return ret;
}
signed main() {
#ifdef JACK
    freopen("NOIPAK++.in","r",stdin);
#endif
    n=g();
    for(R i=1,u,v,w;i<n;++i) u=g(),v=g(),w=g(),add(u,v,w),add(v,u,w);
    dfs(1,0); for(R i=1;i<=n;++i) ins(d[i]); R ans=0;
    for(R i=1;i<=n;++i) ans=max(ans,query(d[i])); printf("%d\n",ans);     
}

2019.06.13

 

Luogu P4551 最长异或路径 01trie

标签:etc   fir   pre   for   map   max   ace   不同   不同的   

原文地址:https://www.cnblogs.com/Jackpei/p/11013652.html

(0)
(0)
   
举报
评论 一句话评论(0
0条  
登录后才能评论!
© 2014 mamicode.com 版权所有 京ICP备13008772号-2
迷上了代码!