BZOJ1086 [SCOI2005]王室联邦（树分块）

把树的结点分块，块内结点连通且个数[b,3b]。

一遍DFS，维护一个栈，设置一个虚拟栈底以保证连通，递归返回时判断栈内元素个数是否大于等于b，是则划分为一个块，最后剩下的与最后一个块划分在一起。

http://blog.csdn.net/popoqqq/article/details/42772237

 #include<cstdio>
 #include<cstring>
 using namespace std;
 #define MAXN 1111
 struct Edge{
     int v,next;
 }edge[MAXN<<];
 int NE,head[MAXN];
 void addEdge(int u,int v){
     edge[NE].v=v; edge[NE].next=head[u];
     head[u]=NE++;
 }
 int b;
 int root[MAXN],belong[MAXN],bn;
 int stack[MAXN],top;
 void dfs(int u,int fa){
     int buttom=top;
     for(int i=head[u]; i!=-; i=edge[i].next){
         int v=edge[i].v;
         if(v==fa) continue;
         dfs(v,u);
         if(top-buttom>=b){
             root[++bn]=u;
             while(top!=buttom){
                 belong[stack[top--]]=bn;
             }
         }
     }
     stack[++top]=u;
 }
 int main(){
     int n,u,v;
     scanf("%d%d",&n,&b);
     NE=;
     memset(head,-,sizeof(head));
     for(int i=; i<n; ++i){
         scanf("%d%d",&u,&v);
         addEdge(u,v);
         addEdge(v,u);
     }
     dfs(,);
     if(top && bn==){
         putchar('');
         return ;
     }
     while(top){
         belong[stack[top--]]=bn;
     }
     printf("%d\n",bn);
     for(int i=; i<=n; ++i) printf("%d ",belong[i]);
     putchar('\n');
     for(int i=; i<=bn; ++i) printf("%d ",root[i]);
     return ;
 }