想学树上莫队结果做了个树分块。
看完题后想到了菊花图的形状认为无解,结果仔细一瞧省会可以在外省尴尬
对于每一颗子树进行深搜,一旦遇到加在一起大小达到B则将它们并为一省,因为他子树搜完以后没有分出块的大小是小于B的,而且他自己当前剩下的也是小于B的,所以可以放心和。
最后剩下的点肯定也小于B所以与最后一个省一合即可。当前子树根节点留给上面也是因为这个。
By:大奕哥
#include<bits/stdc++.h>
using namespace std;
const int N=;
int bel[N],size[N],cap[N],head[N],num,cnt,top,s[N],n,b;
struct node{
int to,nex;
}e[N<<];
void add(int x,int y)
{
e[++cnt].nex=head[x];e[cnt].to=y;head[x]=cnt;
}
void dfs(int x,int fa)
{
s[++top]=x;
for(int i=head[x];i;i=e[i].nex)
{
int y=e[i].to;
if(y==fa)continue;
dfs(y,x);
if(size[x]+size[y]>=b){
size[x]=;
cap[++num]=x;
while(s[top]!=x)bel[s[top--]]=num;
}
else size[x]+=size[y];
}
size[x]++;
}
int main()
{
scanf("%d%d",&n,&b);
if(n<b){
puts("");return ;
}
int x,y,w;
for(int i=;i<n;++i)
{
scanf("%d%d",&x,&y);
add(x,y);add(y,x);
}
dfs(,);
while(top)bel[s[top--]]=num;
printf("%d\n",num);
for(int i=;i<=n;++i)printf("%d ",bel[i]);
printf("\n");
for(int i=;i<=num;++i)printf("%d ",cap[i]);
return ;
}