牛客國慶集訓派對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;
}