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<<1];
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!=-1; 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=0;
    memset(head,-1,sizeof(head));
    for(int i=1; i<n; ++i){
        scanf("%d%d",&u,&v);
        addEdge(u,v);
        addEdge(v,u);
    }
    dfs(1,1);
    if(top && bn==0){
        putchar('0');
        return 0;
    }
    while(top){
        belong[stack[top--]]=bn;
    }
    printf("%d
",bn);
    for(int i=1; i<=n; ++i) printf("%d ",belong[i]);
    putchar('
');
    for(int i=1; i<=bn; ++i) printf("%d ",root[i]);
    return 0;
}