牛客国庆集训派对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;
}