牛客国庆集训派对Day6 I 清明梦超能力者黄YY

清明梦超能力者黄YY

传送门

题目：

黄YY是一个清明梦超能力者，同时也是一个记忆大师。他能够轻松控制自己在梦中的一切，在醒来之后还能清晰的记得梦中所有的细节，这让他的朋友们都十分羡慕。
又是一个晚上，黄YY又到了自己的梦中，并且随手造出了一棵有n个点的树，树上每个点有一个初始颜色0。为了让这棵树不那么单调，黄YY拿起了画笔在上面尽情上色。每一次上色可以用u,
v, c来描述，代表黄YY把u, v这条路径上的点都染色成了c。
正当黄YY开心的完成了m次染色，准备在早上醒来之时向朋友们炫耀。但现实中的黄YY由于过于兴奋滚到了床下，撞到了脑袋，在剧痛中醒来。由于脑部受到了严重创伤，黄YY对刚才梦境中发生的一切发生了严重的信息丢失。
但英俊潇洒的黄YY当然不希望自己的窘态被朋友们发现。为了证明自己还是那个清明梦超能力者，他希望告诉朋友们自己上色后每个节点的颜色。同时为了更进一步证明他还是个记忆大师，他希望干脆直接说出每个点在倒数第k次染色时的颜色。
当然，现在的黄YY已经成了弱智了，作为黄YY最亲密的朋友，你快来帮帮黄YY吧！

题解：

这题可以用树链剖分做。染色的时候就用树链剖分修改操作。因为是要求倒数第K次的染色情况，因此我们可以倒着染色，然后初始化染色次数为K，每次操作就K–，并用线段树维护区间最小值，一旦出现最小值就把这个点加到答案上。

代码：

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
const int inf=0x3f3f3f;
struct edge
{
	int next,to;
}e[maxn<<1];
int head[maxn],cnt;
void add(int u,int v)
{
	cnt++;
	e[cnt].to=v;
	e[cnt].next=head[u];
	head[u]=cnt;
}
int siz[maxn],fa[maxn],son[maxn],rnk[maxn],dep[maxn],num[maxn],top[maxn],tot;

void dfs1(int u)
{
	siz[u]=1;
	for(int i=head[u];i;i=e[i].next)
	{
//		cout<<"----"<<endl;
		if(e[i].to==fa[u])continue;
		dep[e[i].to]=dep[u]+1;
		fa[e[i].to]=u;
		dfs1(e[i].to);
		siz[u]+=siz[e[i].to];
		if(siz[e[i].to]>siz[son[u]]) son[u]=e[i].to;
	}
 } 
void dfs2(int x,int tp)
{
	top[x]=tp;
	num[x]=++tot;
	rnk[tot]=x;
	if(son[x])dfs2(son[x],tp);
	for(int i=head[x];i;i=e[i].next)
		if(e[i].to!=fa[x]&&e[i].to!=son[x])
			dfs2(e[i].to,e[i].to); 
}
struct node
{
	int l,r,c;
}opt[maxn];

int tree[maxn<<1],ls[maxn<<1],rs[maxn<<1],lz[maxn<<1],trtot;
int ans[maxn],Color;
int n,m,K;

void pushup(int k,int l,int r)
{
	if(l==r)return;
	tree[k]=min(tree[ls[k]],tree[rs[k]]);
}
void pushdown(int k,int l,int r)
{
	if(l==r)return;
	if(!lz[k])return;
	int ff=lz[k],ll=ls[k],rr=rs[k];
	tree[ll]-=ff,tree[rr]-=ff;
	lz[ll]+=ff,lz[rr]+=ff,lz[k]=0;
}
void build(int k,int l,int r)
{
	if(l==r)
	{
		tree[k]=K;
		return;
	}
	int mid=(l+r)>>1;
	ls[k]=++trtot;
	rs[k]=++trtot;
	build(ls[k],l,mid);
	build(rs[k],mid+1,r);
	pushup(k,l,r);
}
void change(int k,int l,int r,int x,int y)
{
	pushdown(k,l,r);
//	cout<<k<<" "<<ls[k]<<" "<<rs[k]<<endl;
	if(l==x&&r==y)
	{
		tree[k]--,lz[k]++;
		return;
	}
	int mid=(l+r)>>1;
	if(y<=mid) 
		change(ls[k],l,mid,x,y);
	else if(x>mid) 
		change(rs[k],mid+1,r,x,y);
	else change(ls[k],l,mid,x,mid),change(rs[k],mid+1,r,mid+1,y);
	pushup(k,l,r);
}
void modify(int k,int l,int r)
{
	pushdown(k,l,r);
	if(l==r&&tree[k]==0)
	{
		tree[k]=inf;
		ans[rnk[l]]=Color;
		return;
	}
	int mid=(l+r)>>1;
	if(!tree[ls[k]]) modify(ls[k],l,mid);
	if(!tree[rs[k]]) modify(rs[k],mid+1,r);
	pushup(k,l,r);
}
void update_path(int x,int y,int c)
{
	Color = c;
	int xx=top[x],yy=top[y];
	while(xx!=yy)
	{
//		cout<<"------"<<endl;
		if(dep[xx]>dep[yy])
		{
			change(1,1,n,num[xx],num[x]);
			x=fa[xx];
		}
		else
		{
			change(1,1,n,num[yy],num[y]);
			y=fa[yy];
		}
		xx=top[x];
		yy=top[y];
	}
	if(x!=y)
	{
		if(num[x]>num[y])swap(x,y);
		change(1,1,n,num[x],num[y]);
	}
	else change(1,1,n,num[x],num[x]);
	modify(1,1,n);
}
int main()
{
	cin>>n>>m>>K;
	for(int i=1;i<n;i++)
	{
		int u,v;
		scanf("%d%d",&u,&v);
		add(u,v);add(v,u);
	}
	dfs1(1);
	dfs2(1,1);
	trtot=1;
	build(1,1,n);
//	cout<<"------------"<<endl;
	for(int i=1;i<=m;i++)
		scanf("%d%d%d",&opt[i].l,&opt[i].r,&opt[i].c);
	
	for(int i=m;i;i--)
		update_path(opt[i].l,opt[i].r,opt[i].c);
	for(int i=1;i<=n;i++)
		printf("%d%c",ans[i],i==n?'\n':' ');
	return 0;
}