题意
国王想重新编制他的国家。他想把他的国家划分成若干个省。他的国家有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;
}