萌新求助为啥我的trie树莫名tle啊,做了两道题都是
  • 板块学术版
  • 楼主shitbro
  • 当前回复2
  • 已保存回复2
  • 发布时间2020/6/23 16:21
  • 上次更新2023/11/7 00:11:33
查看原帖
萌新求助为啥我的trie树莫名tle啊,做了两道题都是
90972
shitbro楼主2020/6/23 16:21

rt 这是最长异或路径

#include<bits/stdc++.h>
#define MAXN 400005
using namespace std;
struct node {
	int nxt,to,val;
}t[MAXN];
int tr=0,head[MAXN],ch[MAXN*10][5],cnt[MAXN],tot=0,maxl=0;
void add(int u,int v,int w) {
	t[++tr].nxt=head[u];
	t[tr].to=v;
	t[tr].val=w;
	head[u]=tr;
}
void dfs(int u,int fa) {
	for(int i=head[u];i;i=t[i].nxt) {
		int v=t[i].to;
		if(v==fa) continue;
		cnt[v]=cnt[u]^t[i].val;
		dfs(v,u);
	}
}
void build(int x) {
	int num[MAXN]={0},u=0;
	for(int i=(1<<30);i;i>>=1){
		bool c=i&x;
		if(!ch[u][c]) ch[u][c]=++tot;
		u=ch[u][c];
	}
}
int query(int x) {
	int ans=0,u=0;
	for(int i=(1<<30);i;i>>=1){
		bool c=i&x;
		if(ch[u][!c]) {
			ans+=i;
			u=ch[u][!c];
		}
		else u=ch[u][c];
	}
	return ans;
}
int main()
{
	freopen("2.in","r",stdin);
	freopen("2.out","w",stdout);
	int n; scanf("%d",&n);
	for(int i=1;i<n;i++) {
		int u,v,w; scanf("%d%d%d",&u,&v,&w);
		add(u,v,w); add(v,u,w);
	}
	dfs(1,0);
	for(int i=1;i<=n;i++) build(cnt[i]);
	for(int i=1;i<=n;i++) maxl=max(maxl,query(cnt[i]));
	printf("%d\n",maxl);
	return 0;
}

2020/6/23 16:21
加载中...