[树分块] 1086: [SCOI2005]王室联邦

题意

国王想重新编制他的国家。他想把他的国家划分成若干个省。他的国家有n个城市，编号为1..n。
一些城市之间有道路相连，任意两个不同的城市之间有且仅有一条直接或间接的道路。
每个省至少要有B个城市，每个省最多只有3B个城市。
每个省必须有一个省会，这个省会可以位于省内，也可以在该省外。但是该省的任意一个城市到达省会所经
过的道路上的城市（除了最后一个城市，即该省省会）都必须属于该省。一个城市可以作为多个省的省会。

题解

学树上莫队的前置知识。
感觉这并不是一道题目，就是让你学会树分块是怎么分的……
如何搞呢？
%%% PoPoQQQ http://blog.csdn.net/popoqqq/article/details/42772237
维护一个栈，表示等待被划分的元素。
从根开始dfs，每个点退出递归时压栈。
在每次dfs(x)刚开始时，先记下当前的top作为限制，之后在x的子树搞事情的时候把他当成栈底，再下面的元素就不要动了。
如果某棵子树深dfs后栈内元素数大于等于B，就把当前栈内元素（除去限制之下的）全部pop并把他们分成一个块。
为什么这么搞呢？因为这样保证了分在同一块的节点的距离尽量近，不会出现类似几个距离很远的叶子节点分到一个块的情况。
显然这样分出的块的大小都小于2B。
深搜结束后栈中可能会剩余一些节点。我们把他们分到之前已经分好的最后一块中，可以保证最后一块的大小不超过3b。
还有省会节点。其实只要当每次集体弹栈时把这块的省会几位现在的x即可。
具体看代码吧。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100005, maxe=200005;
int n,B,blg[maxn],G,stk[maxn],rt[maxn],top,fir[maxn],nxt[maxe],son[maxe],tot;
void add(int x,int y){
    son[++tot]=y; nxt[tot]=fir[x]; fir[x]=tot;
}
void dfs(int x,int pre){
    int bottom=top;
    for(int j=fir[x];j;j=nxt[j]) if(son[j]!=pre){
        dfs(son[j],x);  
        if(top-bottom>=B){  
            rt[++G]=x; 
            while(top!=bottom) blg[stk[top--]]=G;
        }
    }
    stk[++top]=x; 
} 
int main(){
    scanf("%d%d",&n,&B);
    for(int i=1;i<=n-1;i++){
        int x,y; scanf("%d%d",&x,&y);
        add(x,y); add(y,x);
    }
    dfs(1,0);
    while(top) blg[stk[top--]]=G;
    if(!G){ printf("0\n"); return 0; }
    printf("%d\n",G);
    for(int i=1;i<=n;i++) printf("%d%c",blg[i],(i==n?'\n':' '));
    for(int i=1;i<=G;i++) printf("%d%c",rt[i],(i==G?'\n':' '));
    return 0; 
}