BZOJ 1086 王室联邦 [王室联邦-树上分块]

这是一种特殊的分块方法，具体情况我论文中有讲述。
建议大家寻常的树上分块可以使用这种分块方法，不容易被卡

#include<bits/stdc++.h>
using namespace std;
const int N = 10000;
inline void read(int &res){
    static char ch;int flag=1;
    while((ch=getchar())<'0'||ch>'9')if(ch=='-')flag=-1;res=ch-48;
    while((ch=getchar())>='0'&&ch<='9')res=res*10+ch-48;res*=flag;
}
int n,m,cnt,top,pro;
int last[N],q[N],sz[N],sum[N],belong[N];
struct data{int to,next;}E[2005];
void insert(int u,int v){
    E[++cnt].to=v;E[cnt].next=last[u];last[u]=cnt;
    E[++cnt].to=u;E[cnt].next=last[v];last[v]=cnt;
}
void dfs(int x,int fa){
    q[++top]=x;
    for(int i=last[x];i;i=E[i].next)if(E[i].to!=fa){
        dfs(E[i].to,x);
        if(sz[x]+sz[E[i].to]>=m){
            sz[x]=0;
            sum[++pro]=x;
            while(q[top]!=x)
                belong[q[top--]]=pro;
        }else sz[x]+=sz[E[i].to];
    }
    sz[x]++;
}
void paint(int x,int fa,int c){
    if(belong[x])c=belong[x];
    else belong[x]=c;
    for(int i=last[x];i;i=E[i].next)
        if(E[i].to!=fa)
            paint(E[i].to,x,c);
}
int main() 
{
    read(n),read(m);
    if(n<m){
        puts("0");return 0;
    }
    for(register int u,v,i=1;i<n;i++)
        read(u),read(v),insert(u,v);
    dfs(1,0);
    if(!pro)sum[++pro]=1;
    paint(1,0,pro);
    printf("%d\n",pro);
    for(register int i=1;i<=n;i++)printf("%d ",belong[i]);
    printf("\n");
    for(register int i=1;i<=pro;i++)printf("%d ",sum[i]);
    printf("\n");
    return 0;
}