传送门biu~
从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;
}