bzoj 1086 [SCOI2005]王室联邦

http://www.elijahqi.win/archives/3440
Description
　　“余”人国的国王想重新编制他的国家。他想把他的国家划分成若干个省，每个省都由他们王室联邦的一个成
员来管理。他的国家有n个城市，编号为1..n。一些城市之间有道路相连，任意两个不同的城市之间有且仅有一条
直接或间接的道路。为了防止管理太过分散，每个省至少要有B个城市，为了能有效的管理，每个省最多只有3B个
城市。每个省必须有一个省会，这个省会可以位于省内，也可以在该省外。但是该省的任意一个城市到达省会所经
过的道路上的城市（除了最后一个城市，即该省省会）都必须属于该省。一个城市可以作为多个省的省会。聪明的
你快帮帮这个国王吧！

Input
　　第一行包含两个数N，B（1<=N<=1000, 1 <= B <= N）。接下来N－1行，每行描述一条边，包含两个数，即这
条边连接的两个城市的编号。

Output
　　如果无法满足国王的要求，输出0。否则输出数K，表示你给出的划分方案中省的个数，编号为1..K。第二行输
出N个数，第I个数表示编号为I的城市属于的省的编号，第三行输出K个数，表示这K个省的省会的城市编号，如果
有多种方案，你可以输出任意一种。

Sample Input
8 2
1 2
2 3
1 8
8 7
8 6
4 6
6 5
Sample Output
3
2 1 1 3 3 3 3 2
2 1 8
HINT

Source
为什么限制是保证块大小在B&3B以内 因为树分块的特性 我们考虑如果B一块的分的话如果直接按照数列分块去写的话 显然有一种问题 就是可能分完之后导致整个块不连通了 那么我们考虑改进一下分块的方式 在每个节点为根节点的时候对其子树进行分块 并且设置一个bot 保证分块的时候不会把我上一层的节点分掉 分块的时候就是循环完子树 然后如果我这个子树已经超过了B 肯定直接在里面就分完了 考虑一种情况就是我在一个子树标号之后发现并没有够B个 那么我再下到其中一个子树的时候发现也没有超过B但是两个这样的块拼起来就是最多不超过2*B 这样分块也可以扩展到树上莫队我们考虑这样分块之后右端点按照dfs序排序 这样的话可以保证我移动的时候最多移动2*n个节点 所以复杂度还是Okay 的那么继续回到这题来我们分块的时候有可能最后剩下了不足B个点 那么我们把他们全部都分到最后一块即可 所以最后保证块的大小其实是小于3B的

#include<cstdio>
#include<cctype>
#include<algorithm>
using namespace std;
inline char gc(){
    static char now[1<<16],*S,*T;
    if (T==S){T=(S=now)+fread(now,1,1<<16,stdin);if (T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(!isdigit(ch)) {if(ch=='-') f=-1;ch=gc();}
    while(isdigit(ch)) x=x*10+ch-'0',ch=gc();
    return x*f;
}
const int N=1e3+10;
struct node{
    int y,next;
}data[N<<1];
int q[N],top,cnt,id[N],b,num,h[N],n,cap[N];
inline void dfs(int x,int bot,int fa){
    for (int i=h[x];i;i=data[i].next){
        int y=data[i].y;if (y==fa) continue;
        dfs(y,top,x);
        if (top-bot>=b){
            ++cnt;cap[cnt]=x;
            while(top>bot) id[q[top--]]=cnt;
        }
    }q[++top]=x;
}
int main(){
    //freopen("bzoj1086.in","r",stdin);
    n=read();b=read();
    for (int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].y=y;data[num].next=h[x];h[x]=num;
        data[++num].y=x;data[num].next=h[y];h[y]=num;
    }dfs(1,0,1);
    while(top>0) id[q[top--]]=cnt;  
    printf("%d\n",cnt);
    for (int i=1;i<=n;++i) printf("%d ",id[i]);puts("");
    for (int i=1;i<=cnt;++i) printf("%d ",cap[i]);
    return 0;
}