BZOJ 1086([SCOI2005]王室联邦-树的划分)

题意：国家有n个城市，编号为1..n。一些城市之间有道路相连，任意两个不同的城市之间有且仅有一条直接或间接的道路。为了防止管理太过分散，每个省至少要有B个城市，为了能有效的管理，每个省最多只有3B个城市。每个省必须有一个省会，这个省会可以位于省内，也可以在该省外。但是该省的任意一个城市到达省会所经过的道路上的城市（除了最后一个城市，即该省省会）都必须属于该省。求任意可行解。n,b<=1000

dfs遍历树，当出现一个节点数>=b的子树时，以该节点为省会，取掉前k个子树（前k-1个子树节点数和均小于b，且加上节点数

#include<bits/stdc++.h>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define ForkD(i,k,n) for(int i=n;i>=k;i--)
#define RepD(i,n) for(int i=n;i>=0;i--)
#define Forp(x) for(int p=Pre[x];p;p=Next[p])
#define Forpiter(x) for(int &p=iter[x];p;p=Next[p]) 
#define Lson (o<<1)
#define Rson ((o<<1)+1)
#define MEM(a) memset(a,0,sizeof(a));
#define MEMI(a) memset(a,127,sizeof(a));
#define MEMi(a) memset(a,128,sizeof(a));
#define INF (2139062143)
#define F (100000007)
#define pb push_back
#define mp make_pair 
#define fi first
#define se second
#define vi vector<int> 
#define pi pair<int,int>
#define SI(a) ((a).size())
#define Pr(kcase,ans) printf("Case %d: %lld\n",kcase,ans);
#define PRi(a,n) For(i,n-1) cout<<a[i]<<' '; cout<<a[n]<<endl;
#define PRi2D(a,n,m) For(i,n) { \
                        For(j,m-1) cout<<a[i][j]<<' ';\
                        cout<<a[i][m]<<endl; \
                        } 
typedef long long ll;
typedef unsigned long long ull;
ll mul(ll a,ll b){return (a*b)%F;}
ll add(ll a,ll b){return (a+b)%F;}
ll sub(ll a,ll b){return (a-b+llabs(a-b)/F*F+F)%F;}
void upd(ll &a,ll b){a=(a%F+b%F)%F;}
int read()
{
    int x=0,f=1; char ch=getchar();
    while(!isdigit(ch)) {if (ch=='-') f=-1; ch=getchar();}
    while(isdigit(ch)) { x=x*10+ch-'0'; ch=getchar();}
    return x*f;
} 
const int MAXN = 1010;
int n,B;
vector<int> G[MAXN];
int s[MAXN],siz=0;
int cnt[MAXN]={0},c=0,rot[MAXN];
void dfs(int x,int fa) {
    int lim=siz;
    int m=SI(G[x]);
    Rep(i,m) {
        int v=G[x][i];
        if (v==fa) continue;
        dfs(v,x);
        if (siz-lim>=B) {
            ++c;
            while(lim<siz) cnt[s[siz--]]=c;
            rot[c]=x;
        }
    }
    s[++siz]=x;
}
int main()
{
// freopen("bzoj1086.in","r",stdin);
// freopen(".out","w",stdout);

    cin>>n>>B;
    For(i,n-1) {
        int a=read(),b=read();
        G[a].pb(b);
        G[b].pb(a);
    }
    dfs(1,0); 
    while(siz) cnt[s[siz--]]=c;
    cout<<c<<endl;
    PRi(cnt,n)
    PRi(rot,c)
    return 0;
}