CF980E The Number Games

一、题目

点此看题

二、解法

最重要的条件就是：第i个省出来的代表有2^i名粉丝，我们不能贪心删最小的叶子，而应该贪心倾尽全力保留编号最大的点。

我们要维护一个联通块，$n$点肯定在其中，那我们可以$n$为根建这棵树，考虑从大到小加入点，一个点被加入联通块就需要它到根的路径都被加入联通块，我们查询它到根路径上未被加入的点数，看都加入会不会爆$k$的限制。

别看数据$1e6$，树链剖分猛如虎！然后我们就愉快的$\text{AC}$了本题，贴个代码$qwq$。

#include <cstdio>
const int M = 1000005;
int read()
{
 int x=0,flag=1;char c;
 while((c=getchar())<'0' || c>'9') if(c=='-') flag=-1;
 while(c>='0' && c<='9') x=(x<<3)+(x<<1)+(c^48),c=getchar();
 return x*flag;
}
int n,m,tot,f[M],dep[M],siz[M],son[M],top[M];
bool fuck[M];int now,Index,fa[M],tr[4*M],num[M];
struct edge
{
    int v,next;
    edge(int V=0,int N=0) : v(V) , next(N) {}
}e[2*M];
void dfs1(int u,int p)
{
    fa[u]=p;
    siz[u]=1;
    dep[u]=dep[p]+1;
    for(int i=f[u];i;i=e[i].next)
    {
        int v=e[i].v;
        if(v==p) continue;
        dfs1(v,u);
        if(siz[v]>siz[son[u]])
            son[u]=v;
        siz[u]+=siz[v];
    }
}
void dfs2(int u,int tp)
{
    top[u]=tp;
    num[u]=++Index;
    if(son[u]) dfs2(son[u],tp);
    for(int i=f[u];i;i=e[i].next)
        if(e[i].v^fa[u] && e[i].v^son[u])
            dfs2(e[i].v,e[i].v);
}
void upd(int i,int l,int r,int id)
{
    tr[i]++;
    if(l==r) return ;
    int mid=(l+r)>>1;
    if(mid>=id)
        upd(i<<1,l,mid,id);
    else
        upd(i<<1|1,mid+1,r,id);
}
int query(int i,int l,int r,int L,int R)
{
    if(l>R || L>r) return 0;
    if(L<=l && r<=R) return tr[i];
    int mid=(l+r)>>1;
    return query(i<<1,l,mid,L,R)+query(i<<1|1,mid+1,r,L,R);
}
int ask(int x)
{
    int ans=0;
    while(top[x]^n)
    {
        ans+=query(1,1,n,num[top[x]],num[x]);
        x=fa[top[x]];
    }
    ans+=query(1,1,n,1,num[x]);
    return ans;
}
void work(int i)
{
    if(dep[i]-ask(i)+now>m) return ;
    for(;i && !fuck[i];i=fa[i])
    {
        fuck[i]=1;now++;
        upd(1,1,n,num[i]);
    }
}
int main()
{
    n=read();m=n-read();
    for(int i=1;i<n;i++)
    {
        int u=read(),v=read();
        e[++tot]=edge(v,f[u]),f[u]=tot;
        e[++tot]=edge(u,f[v]),f[v]=tot;
    }
    dfs1(n,0);
    dfs2(n,n);
    for(int i=n;i>=1;i--)
        if(!fuck[i])
            work(i);
    for(int i=1;i<=n;i++)
        if(!fuck[i])
            printf("%d ",i);
}