BZOJ1086 SCOI2005王室联邦

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

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

对于每一颗子树进行深搜，一旦遇到加在一起大小达到B则将它们并为一省，因为他子树搜完以后没有分出块的大小是小于B的，而且他自己当前剩下的也是小于B的，所以可以放心和。

最后剩下的点肯定也小于B所以与最后一个省一合即可。当前子树根节点留给上面也是因为这个。

By：大奕哥

#include<bits/stdc++.h>
using namespace std;
const int N=1005;
int bel[N],size[N],cap[N],head[N],num,cnt,top,s[N],n,b;
struct node{
    int to,nex;
}e[N<<1];
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]=0;
            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("0");return 0;
    }
    int x,y,w;
    for(int i=1;i<n;++i)
    {
        scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    dfs(1,0);
    while(top)bel[s[top--]]=num;
    printf("%d\n",num);
    for(int i=1;i<=n;++i)printf("%d ",bel[i]);
    printf("\n");
    for(int i=1;i<=num;++i)printf("%d ",cap[i]);
    return 0;
}