2018牛客国庆集训派对Day6 I 清明梦超能力者黄YY(树链剖分)

传送门

题意：中文

思路：既然求倒数第k次染为什么颜色，那就倒着染色，染到第k次就标记下。
所以我们可以用线段树来维护每个点被更新的次数，维护一条链上的节点被更新过的次数的最大次数。当一个区间的最大被更新次数超过k时，我们再暴力往下去找具体区间，对答案进行更新，对一个节点的答案更新完之后就将其的值设为负无穷大。由于每个点算答案的时候都只会被算到一次，所以整体的复杂度应该是 O(n*logn*logn)。

注意idx记录线段树节点所对应的原本节点号
 

#include<bits/stdc++.h>
using namespace std;
const int maxn = 1e5+5;
const int INF  = 0x3f3f3f3f;
int a[maxn],idx[maxn],ans[maxn];
int n,m,k;
vector<int> G[maxn];
struct Qes{
    int u,v,c;
}q[maxn];

//segment tree
#define ls (k1<<1)
#define rs (k1<<1|1)
#define mid (l+r>>1)
int sum[maxn<<2],lazy[maxn<<2];
void pushup(int k1){sum[k1]=min(sum[ls],sum[rs]);}
void build(int k1,int l,int r){
    sum[k1]=k;
    if(l==r) {
        return ;
    }
    build(ls,l,mid);
    build(rs,mid+1,r);
    //pushup(k1);
}
void pushdown(int k1){
    if(lazy[k1]){
        lazy[ls] += lazy[k1];
        lazy[rs] += lazy[k1];
        sum[ls]  -= lazy[k1];
        sum[rs]  -= lazy[k1];
        lazy[k1] =  0;
    }
}
// single point
void query(int k1,int l,int r,int c){
    if(sum[k1]>0) return ;
    if(l==r){
        ans[idx[l]]=c;
        sum[k1]=INF;
        return ;
    }
    pushdown(k1);
    query(ls,l,mid,c);
    query(rs,mid+1,r,c);
    pushup(k1);
}
void update(int k1,int l,int r,int L,int R,int c){
    if(L<=l && r<=R){
        sum[k1]--;
        lazy[k1]++;
        query(k1,l,r,c);
        return ;
    }
    pushdown(k1);
    if(L<=mid) update(ls,l,mid,L,R,c);
    if(R >mid) update(rs,mid+1,r,L,R,c);
    pushup(k1);

}
//hexvy
int dep[maxn],fa[maxn],sz[maxn],son[maxn];
int id[maxn],top[maxn],cnt,wt[maxn],w[maxn];
void dfs1(int u,int f,int deep){
    dep[u]=deep;
    fa[u]=f;
    sz[u]=1;
    son[u]=0;
    for(int v : G[u]){
        if(v==f) continue;
        dfs1(v,u,deep+1);
        sz[u]+=sz[v];
        if(sz[v]>sz[son[u]]) son[u]=v;
    }
}
void dfs2(int u,int topf){
    id[u]=++cnt;
    idx[cnt]=u;
    //wt[cnt]=w[u];//init 0
    top[u]=topf;
    if(!son[u]) return ;
    dfs2(son[u],topf);
    for(int v : G[u]){
        if(v==fa[u] || v==son[u]) continue;
        dfs2(v,v);
    }
}
void addRange(int x,int y,int c){
    while(top[x]!=top[y]){
        if(dep[top[x]]<dep[top[y]]) swap(x,y);
        update(1,1,n,id[top[x]],id[x],c);
        x=fa[top[x]];
    }
    if(dep[x]>dep[y]) swap(x,y);
    update(1,1,n,id[x],id[y],c);
}
void work(){
    dfs1(1,0,1);
    dfs2(1,1);
    build(1,1,n);

    int u,v,c;
    for(int i=m;i>=1;i--){
        addRange(q[i].u,q[i].v,q[i].c);
    }

    for(int i=1;i<n;i++){
        printf("%d ",ans[i]);
    }
    printf("%d\n",ans[n]);
}

int main(){
    //freopen("in.txt","r",stdin);
    scanf("%d%d%d",&n,&m,&k);
    for(int u,v,i=1;i<n;i++) {
        scanf("%d%d",&u,&v);
        G[u].push_back(v);
        G[v].push_back(u);
    }
    for(int i=1;i<=m;i++){
        scanf("%d%d%d",&q[i].u,&q[i].v,&q[i].c);
    }
    work();
    return 0;
}