自己调了3h了,给hack也行()
#include <bits/stdc++.h>
using namespace std;
#define int long long
const int N=5e4+10,INF=0x3f3f3f3f3f3f3f3f;
int n,m;
struct edge{
int v,nxt;
}e[N*2];
int head[N],cnt=2;
void add(int u,int v){
e[cnt].v=v;
e[cnt].nxt=head[u];
head[u]=cnt++;
}
int fa[N],dep[N],siz[N],son[N];
int dfn[N],top[N],num=0;
void dfs1(int u,int fat){
fa[u]=fat;
dep[u]=dep[fat]+1;
siz[u]=1;
int mx=-1;
for(int i=head[u];i;i=e[i].nxt){
if(e[i].v==fat)continue;
dfs1(e[i].v,u);
siz[u]+=siz[e[i].v];
if(siz[e[i].v]>mx){
mx=siz[e[i].v];
son[u]=e[i].v;
}
}
}
void dfs2(int u,int topf){
dfn[u]=++num;
top[u]=topf;
if(!son[u])return;
dfs2(son[u],topf);
for(int i=head[u];i;i=e[i].nxt){
if(!dfn[e[i].v])
dfs2(e[i].v,e[i].v);
}
}
struct node{
int l,r;
int mn,lazy;
}t[N*4];
void SegPushdown(int p){
t[p*2].mn=min(t[p*2].mn,t[p].mn);
t[p*2+1].mn=min(t[p*2+1].mn,t[p].mn);
}
void SegPushup(int p){
t[p].mn=max(t[p*2].mn,t[p*2+1].mn);
}
void SegBuild(int p,int l,int r){
t[p].l=l,t[p].r=r;
if(l==r){
t[p].mn=INF;
return;
}
int m=l+r>>1;
SegBuild(p*2,l,m);
SegBuild(p*2+1,m+1,r);
SegPushup(p);
}
int SegQuery(int p,int pos){
if(t[p].l==pos&&t[p].r==pos)return t[p].mn;
SegPushdown(p);
int m=t[p].l+t[p].r>>1;
if(pos<=m)return SegQuery(p*2,pos);
if(pos>m)return SegQuery(p*2+1,pos);
}
void SegUpdate(int p,int l,int r,int mn){
if(l<=t[p].l&&t[p].r<=r){
t[p].mn=min(t[p].mn,mn);
return;
}
int m=t[p].l+t[p].r>>1;
SegPushdown(p);
if(l<=m)SegUpdate(p*2,l,r,mn);
if(r>m)SegUpdate(p*2+1,l,r,mn);
SegPushup(p);
}
void TreeUpdate(int x,int y,int mn){
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
SegUpdate(1,dfn[top[x]],dfn[x],mn);
x=fa[top[x]];
}
if(dep[x]>dep[y])swap(x,y);
SegUpdate(1,dfn[x],dfn[y],mn);
}
int TreeLCA(int x,int y){
while(top[x]!=top[y]){
if(dep[top[x]]<dep[top[y]])swap(x,y);
x=fa[top[x]];
}
if(dep[x]<dep[y])return x;
else return y;
}
int u[N],v[N];
signed main(){
ios::sync_with_stdio(false);
cin.tie(0),cout.tie(0);
cin>>n>>m;
for(int i=1;i<n;i++){
int p,q;
cin>>p>>q;
u[i]=p,v[i]=q;
add(p,q),add(q,p);
}
dfs1(1,0);
dfs2(1,1);
SegBuild(1,1,n);
while(m--){
int p,q,r;
cin>>p>>q>>r;
int lca=TreeLCA(p,q);
if(lca==p){
int sp;
for(int i=head[p];i;i=e[i].nxt){
if(TreeLCA(e[i].v,q)==e[i].v){
sp=e[i].v;
break;
}
}
TreeUpdate(sp,q,r);
}else if(lca==q){
int sq;
for(int i=head[q];i;i=e[i].nxt){
if(TreeLCA(e[i].v,p)==e[i].v){
sq=e[i].v;
break;
}
}
TreeUpdate(sq,p,r);
}else{
int s1,s2;
for(int i=head[lca];i;i=e[i].nxt){
if(TreeLCA(e[i].v,p)==e[i].v){
s1=e[i].v;
}else if(TreeLCA(e[i].v,q)==e[i].v){
s2=e[i].v;
}
}
TreeUpdate(s1,p,r);
TreeUpdate(s2,q,r);
}
}
for(int i=1;i<n;i++){
int ans,x;
if(dep[u[i]]>dep[v[i]])x=u[i];
else x=v[i];
ans=SegQuery(1,dfn[x]);
if(ans==INF)cout<<"-1\n";
else cout<<ans<<'\n';
}
return 0;
}