【bzoj1086】王室联邦

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

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

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

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