[bzoj1086]王室联邦

有这样一个贪心的思路：当某一个点搜完某个儿子后，发现当前子树中没有被选入其他省的点数超过了B，就将其当做一个省，并把这个点作为省会（注意：这个点并没有进入这个省），显然可以发现此时每一个省的点数都小于2B。另外，当搜完后发现还有小于B的点，那么就将这些点都归入最后一个省，省的点数仍然小于3B。

#include<bits/stdc++.h>
using namespace std;
#define N 1005
struct ji{
    int nex,to;
}edge[N<<1];
int E,n,m,x,y,head[N],st[N],vis[N],s[N];
void add(int x,int y){
    edge[E].nex=head[x];
    edge[E].to=y;
    head[x]=E++;
}
void dfs(int k,int fa){
    int p=st[0];
    for(int i=head[k];i!=-1;i=edge[i].nex)
        if (edge[i].to!=fa){
            dfs(edge[i].to,k);
            if (st[0]-p>=m){
                s[++s[0]]=k;
                while (p!=st[0])vis[st[st[0]--]]=s[0];
            }
        }
    st[++st[0]]=k;
}
int main(){
    scanf("%d%d",&n,&m);
    memset(head,-1,sizeof(head));
    for(int i=1;i<n;i++){
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);
    }
    dfs(1,0);
    while (st[0])vis[st[st[0]--]]=s[0];
    printf("%d\n",s[0]);
    for(int i=1;i<=n;i++)printf("%d ",vis[i]);
    printf("\n");
    for(int i=1;i<=s[0];i++)printf("%d ",s[i]);
}