CF1296F Berland Beauty

利用贪心的思想进行构造，然后检验。

注意到在一棵树中，一个节点最多只有一个根节点，所以可以把边权转化为点权来存储：一个点与它父节点之间的边的权值，存储在它的身上即可。

5000的数据就是用来暴力跳lca的。

#include <bits/stdc++.h>
using namespace std;
const int N=5e3+5;
int n,u[N],v[N],m,now;
int d[N],p[N][21],col[N];
int cnt,head[N];
struct edge{int next,to;}e[N<<1];
struct number{int u,v,w,lca;}num[N];

inline bool cmp(number a,number b)
{
	return a.w<b.w;	
}
inline void add(int u,int v)
{
	cnt++;
	e[cnt].next=head[u];
	e[cnt].to=v;
	head[u]=cnt;	
}

void dfs(int u,int fa)
{
	for (register int i=1; (1<<i)<=d[u]; ++i) p[u][i]=p[p[u][i-1]][i-1];
	for (register int i=head[u]; i; i=e[i].next)
	if (e[i].to!=fa)
	{
		d[e[i].to]=d[u]+1;
		p[e[i].to][0]=u;
		dfs(e[i].to,u);
	}
}

inline int lca(int a,int b)
{
	if (d[a]>d[b]) swap(a,b);
	for (register int i=20; i>=0; --i) if (d[b]-(1<<i)>=d[a]) b=p[b][i];
	if (a==b) return a;
	for (register int i=20; i>=0; --i) if (p[a][i]!=p[b][i]) a=p[a][i],b=p[b][i];
	return p[a][0]; 
}

int main(){
	scanf("%d",&n);
	for (register int i=1; i<n; ++i) 
	{
		scanf("%d%d",&u[i],&v[i]);
		add(u[i],v[i]);
		add(v[i],u[i]);
	}
	dfs(1,0);
	scanf("%d",&m);
	for (register int i=1; i<=m; ++i) scanf("%d%d%d",&num[i].u,&num[i].v,&num[i].w);
	sort(num+1,num+m+1,cmp);
	for (register int i=1; i<=m; ++i)
	{
		num[i].lca=lca(num[i].u,num[i].v);
		now=num[i].u;
		while (now!=num[i].lca) col[now]=num[i].w,now=p[now][0];
		now=num[i].v;
		while (now!=num[i].lca) col[now]=num[i].w,now=p[now][0];
	}
	for (register int i=1; i<=m; ++i)
	{
		bool jay=true;
		int minn=2e9;
		now=num[i].u;
		while (now!=num[i].lca) 
		{
			minn=min(minn,col[now]);
			now=p[now][0];
		}
		now=num[i].v;
		while (now!=num[i].lca)
		{
			minn=min(minn,col[now]);
			now=p[now][0];
		}
		if (minn!=num[i].w) {printf("%d\n",-1); return 0;}
	}
	for (register int i=1; i<n; ++i)
	{
		if (d[v[i]]<d[u[i]]) swap(u[i],v[i]);
		if (!col[v[i]]) printf("%d ",1); else printf("%d ",col[v[i]]);
	}
return 0;
}