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 ;

 }

巴特西

BZOJ1086 SCOI2005王室联邦

最新文章

热门文章