BZOJ1086 SCOI2005王室联邦

想学树上莫队结果做了个树分块。

看完题后想到了菊花图的形状认为无解，结果仔细一瞧省会可以在外省尴尬

对于每一颗子树进行深搜，一旦遇到加在一起大小达到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 ;
 }