bzoj1086 [SCOI2005]王室联邦（树分块，dfs）

嗯，很好的向你证明了树上分块，每一块大小都可以控制在B~3B之间。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 1010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,B,h[N],num=0,tot=0,bel[N],q[N],top=0,cap[N];
struct edge{
    int to,next;
}data[N<<1];
void dfs(int x,int fa){
    q[++top]=x;int now=top;
    for(int i=h[x];i;i=data[i].next){
        int y=data[i].to;if(y==fa) continue;dfs(y,x);
        if(top-now>=B){cap[++tot]=x;while(top!=now) bel[q[top--]]=tot;}
    }
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();B=read();
    for(int i=1;i<n;++i){
        int x=read(),y=read();
        data[++num].to=y;data[num].next=h[x];h[x]=num;
        data[++num].to=x;data[num].next=h[y];h[y]=num;
    }dfs(1,0);if(!tot) cap[++tot]=1;
    while(top) bel[q[top--]]=tot;
    printf("%d\n",tot);
    for(int i=1;i<=n;++i) printf("%d ",bel[i]);puts("");
    for(int i=1;i<=tot;++i) printf("%d ",cap[i]);puts("");
    return 0;
}