【NOI2017模拟3.25】跳蚤王国

Description

给出一棵n个点的树，你可以对这棵树进行任意次操作，每次操作选择一条边删掉再加上一条边，每次操作结束后要保证这张图仍然是一棵树。
对于每个点i求若要使i成为重心最少需要进行多少次操作。
n<=1e6

Solution

先钦点一个点为根，把操作视为砍掉子树（因为砍掉的可以直接接到重心上）
我们就是要砍掉尽量少的大小<=n/2的子树，并且对于每一个砍完的子树的大小也要<=n/2
那么我们对于每一个点i记录以i为根的子树大小要<=n/2最少要砍多少刀，并且砍完之后剩下的最小的子树大小
我们自底向上进行，对于每一个点我们贪心的将它的所有子树剩余的大小从大到小排序，依次砍掉，直到它的大小<=n/2
然后我们就可以求出根的答案。
再考虑换根，我们把根以上的部分的答案也记录下来，当做一个子树。
接下来的方法和之前是一样的。
不过我们枚举根的子树转移的时候，每次就相当于不能砍掉某一个子树，求剩下的答案。
处理前缀和二分一下就好了。
复杂度O(N log N)，由于常数小可以通过此题。

Code

#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define rep(i,a) for(int i=last[a];i;i=next[i])
using namespace std;
const int N=1e6+5;
int x,y,n,l,tot,ans,size[N],sum[N],an[N];
int t[N*2],next[N*2],last[N];
struct point{int x,sz;}f[N],g[N];
struct note{int x,id;}p[N];
bool cmp(note x,note y) {return x.x>y.x;}
void add(int x,int y) {
    t[++l]=y;next[l]=last[x];last[x]=l;
}
void dfs(int x,int y) {
    size[x]=1;
    rep(i,x)    
        if (t[i]!=y) {
            dfs(t[i],x);
            size[x]+=size[t[i]];
        }
    tot=0;f[x].sz=1;
    rep(i,x) 
        if (t[i]!=y) {
            f[x].x+=f[t[i]].x;
            f[x].sz+=f[t[i]].sz;
            p[++tot].x=f[t[i]].sz;
        }
    sort(p+1,p+tot+1,cmp);
    fo(i,1,tot) {
        if (f[x].sz<=n/2) break;
        f[x].sz-=p[i].x;f[x].x++;
    }
}
int find(int l,int r,int v) {
    while (l<r) {
        int mid=(l+r)/2;
        if (sum[mid]>=v) r=mid;
        else l=mid+1;
    }
    return l;
}
void solve(int x,int y) {
    an[x]=g[x].x;
    rep(i,x) if (t[i]!=y) an[x]+=f[t[i]].x;
    p[tot=1].x=g[x].sz;p[1].id=0;g[x].sz++;
    rep(i,x) 
        if (t[i]!=y) {
            p[++tot].x=f[t[i]].sz;
            p[tot].id=t[i];
            g[x].sz+=f[t[i]].sz;
            g[x].x+=f[t[i]].x;
        }
    sort(p+1,p+tot+1,cmp);
    fo(i,1,tot) sum[i]=sum[i-1]+p[i].x;
    fo(i,1,tot) {
        if (!p[i].id) continue;
        int z,sz=g[x].sz-f[p[i].id].sz;
        if (sz-sum[i-1]<=n/2) {
            z=find(0,i-1,sz-n/2);
            sz=sz-sum[z];
        } else {
            z=find(i+1,tot,sz+f[p[i].id].sz-n/2);
            sz=sz+f[p[i].id].sz-sum[z];z--;
        }
        g[p[i].id].x=g[x].x-f[p[i].id].x+z;
        g[p[i].id].sz=sz;
    } 
    rep(i,x) if (t[i]!=y) solve(t[i],x);
}
int main() {
    freopen("flea.in","r",stdin);
    freopen("flea.out","w",stdout);
    scanf("%d",&n);
    fo(i,1,n-1) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    dfs(1,0);solve(1,0);
    fo(i,1,n) printf("%d\n",an[i]);
}