【bzoj1086】王室联邦

我以为树分块什么的必有高论，结果居然是个暴力……

方法也很简单，我看下每个节点dfs的时候是否已经大于k个，大于的话我就新开一块。

注意dfs的时候当前节点不能放进子树的块中。

#include<bits/stdc++.h>
#define N 100010
using namespace std;
struct Edge{int u,v,next;}G[2*N];
int n,B,top,tot,head[N<<2],cnt,rt[N],block[N],q[N];
inline void addedge(int u,int v){
    G[++tot].u=u;G[tot].v=v;G[tot].next=head[u];head[u]=tot;
    G[++tot].u=v;G[tot].v=u;G[tot].next=head[v];head[v]=tot;
}
void dfs(int u,int f){
    int last=top;
    for(int i=head[u];i;i=G[i].next){
        int v=G[i].v;if(v==f)continue;
        dfs(v,u);
        if(top-last>=B){
            block[++cnt]=u;
            for(;top>last;top--)rt[q[top]]=cnt;
        }
    }
    q[++top]=u;
}
inline int read(){
    int f=1,x=0;char ch;
    do{ch=getchar();if(ch=='-')f=-1;}while(ch<'0'||ch>'9');
    do{x=x*10+ch-'0';ch=getchar();}while(ch>='0'&&ch<='9');
    return f*x;
}
int main(){
    n=read();B=read();
    for(int i=1;i<n;i++){
        int x=read(),y=read();
        addedge(x,y);
    }
    dfs(1,0);
    while(top)rt[q[top--]]=cnt;
    printf("%d\n",cnt);
    for(int i=1;i<=n;i++)printf("%d ",rt[i]);puts("");
    for(int i=1;i<=cnt;i++)printf("%d ",block[i]);puts("");
    return 0;
}