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
思路
从1号节点开始dfs,维护一个栈,每个点退出递归时压栈,自下至上进行合并。如果某棵子树深搜完之后栈内元素数>=b 就把当前的栈内元素合并为一个块。我们在每次进入递归时维护一个栈底,对于当前子树来说这个栈底就是整个栈的底。这样当一棵子树深搜过后由于子树内未分块节点不超过b,之前搜过的未分块节点数也不超过b,因此每块不超过2b。最后将栈里剩下的不足b的节点都加入最后一个块,最后一个块的大小不超过3b,符合题意。
代码
#include<bits/stdc++.h>
using namespace std;
int head[1005],nex[2005],to[2005],tp;
int in[1005],root[1005];
int stac[1005];
int n,b,top,cnt;
void add(int x,int y){
nex[++tp]=head[x];
head[x]=tp;
to[tp]=y;
}
void dfs(int x,int father){
int bottom=top;
for(int i=head[x];i;i=nex[i]){
if(to[i]==father) continue;
dfs(to[i],x);
if(top-bottom>=b){
root[++cnt]=x;
while(top!=bottom) in[stac[top--]]=cnt;
}
}
stac[++top]=x;
}
int main(){
scanf("%d%d",&n,&b);
for(int i=1;i<n;++i){
int a,b;
scanf("%d%d",&a,&b);
add(a,b);add(b,a);
}
dfs(1,0);
while(top) in[stac[top--]]=cnt;
printf("%d\n",cnt);
for(int i=1;i<=n;++i) printf("%d ",in[i]);
printf("\n");
for(int i=1;i<=cnt;++i) printf("%d ",root[i]);
return 0;
}