考虑如果在序列上的话,每次只需要在区间左端点+1,右端点-1,最后用线段树一起统计即可。
在树上的话,就用树链剖分转换成序列,每一次操作,对应修改O(log n)条重链上的标记,最后用线段树来统计。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<iostream>
#include<vector>
#include<map>
#define maxn 200100
using namespace std;
struct yts
{
int l,r;
int mx,tag;
}t[4*maxn];
int head[maxn],st[maxn],to[maxn],next[maxn],seq[maxn],rank[maxn],fa[19][maxn],size[maxn],dd[maxn];
int n,m,num,tot,T,cnt;
int e[maxn],pos[maxn],ans[maxn],dep[maxn];
bool vis[maxn];
vector<pair<int,int> > v[maxn];
map<int,int> mp;
void addedge(int x,int y)
{
num++;to[num]=y;next[num]=st[x];st[x]=num;
}
void dfs(int x)
{
size[x]=1;e[++tot]=x;
for (int p=st[x];p;p=next[p])
if (to[p]!=fa[0][x])
{
fa[0][to[p]]=x;
dep[to[p]]=dep[x]+1;
dfs(to[p]);
size[x]+=size[to[p]];
if (size[to[p]]>size[dd[x]]) dd[x]=to[p];
}
}
int go_up(int x,int d)
{
for (int i=0;i<=17;i++)
if (d&(1<<i)) x=fa[i][x];
return x;
}
int LCA(int x,int y)
{
if (dep[x]>dep[y]) x=go_up(x,dep[x]-dep[y]);
else y=go_up(y,dep[y]-dep[x]);
if (x==y) return x;
for (int i=17;i>=0;i--)
if (fa[i][x]!=fa[i][y]) x=fa[i][x],y=fa[i][y];
return fa[0][x];
}
void add(int x,int a,int b)
{
v[x].push_back(make_pair(a,b));
}
void update(int i)
{
t[i].mx=0;t[i].tag=0;
if (t[i<<1].mx>t[i].mx) t[i].mx=t[i<<1].mx,t[i].tag=t[i<<1].tag;
if (t[i<<1|1].mx>t[i].mx) t[i].mx=t[i<<1|1].mx,t[i].tag=t[i<<1|1].tag;
else if (t[i<<1|1].mx==t[i].mx && t[i].tag>t[i<<1|1].tag) t[i].tag=t[i<<1|1].tag;
}
void build(int i,int l,int r)
{
t[i].l=l;t[i].r=r;t[i].mx=0;
if (l==r) {t[i].tag=pos[l];return;}
int mid=(l+r)/2;
build(i*2,l,mid);build(i*2+1,mid+1,r);
}
void modify(int i,int x,int d)
{
if (t[i].l==t[i].r) {t[i].mx+=d;return;}
int mid=(t[i].l+t[i].r)/2;
if (x<=mid) modify(i*2,x,d);
if (mid<x) modify(i*2+1,x,d);
update(i);
}
int main()
{
scanf("%d%d",&n,&T);
for (int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
addedge(x,y);addedge(y,x);
}
dfs(1);
int qwer=0;
for (int i=1;i<=n;i++)
if (!vis[e[i]])
{
int k=e[i];
while (k)
{
seq[++qwer]=k;vis[k]=1;head[k]=e[i];k=dd[k];
}
}
for (int i=1;i<=n;i++) rank[seq[i]]=i;
for (int j=1;j<=17;j++)
for (int i=1;i<=n;i++)
fa[j][i]=fa[j-1][fa[j-1][i]];
while (T--)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
if (dep[x]<dep[y]) swap(x,y);
if (!mp[z]) mp[z]=++cnt,pos[cnt]=z;
int lca=LCA(x,y);
z=mp[z];
while (dep[head[x]]>dep[lca])
{
add(rank[head[x]],z,1);add(rank[x]+1,z,-1);x=fa[0][head[x]];
}
add(rank[lca],z,1);add(rank[x]+1,z,-1);
if (y!=lca)
{
int w=go_up(y,dep[y]-dep[lca]-1);
while (dep[head[y]]>dep[w])
{
add(rank[head[y]],z,1);add(rank[y]+1,z,-1);y=fa[0][head[y]];
}
add(rank[w],z,1);add(rank[y]+1,z,-1);
}
}
build(1,1,cnt);
for (int i=1;i<=n;i++)
{
for (int j=0;j<v[i].size();j++) modify(1,v[i][j].first,v[i][j].second);
ans[seq[i]]=t[1].tag;
}
for (int i=1;i<=n;i++) printf("%d\n",ans[i]);
return 0;
}