BZOJ1086 [SCOI2005]王室联邦 【dfs + 贪心】

题目

　　“余”人国的国王想重新编制他的国家。他想把他的国家划分成若干个省，每个省都由他们王室联邦的一个成
员来管理。他的国家有n个城市，编号为1..n。一些城市之间有道路相连，任意两个不同的城市之间有且仅有一条
直接或间接的道路。为了防止管理太过分散，每个省至少要有B个城市，为了能有效的管理，每个省最多只有3B个
城市。每个省必须有一个省会，这个省会可以位于省内，也可以在该省外。但是该省的任意一个城市到达省会所经
过的道路上的城市（除了最后一个城市，即该省省会）都必须属于该省。一个城市可以作为多个省的省会。聪明的
你快帮帮这个国王吧！

输入格式

　　第一行包含两个数N，B（1<=N<=1000, 1 <= B <= N）。接下来N－1行，每行描述一条边，包含两个数，即这
条边连接的两个城市的编号。

输出格式

　　如果无法满足国王的要求，输出0。否则输出数K，表示你给出的划分方案中省的个数，编号为1..K。第二行输
出N个数，第I个数表示编号为I的城市属于的省的编号，第三行输出K个数，表示这K个省的省会的城市编号，如果
有多种方案，你可以输出任意一种。

输入样例

8 2

1 2

2 3

1 8

8 7

8 6

4 6

6 5

输出样例

3

2 1 1 3 3 3 3 2

2 1 8

题解

一道长得很像难题的水题。。。【n <= 1000，无解，3B，都是什么奇奇怪怪的东西】
我们只要深搜维护一个栈，当子树剩余没有分配的城市大小>=B时，就将它们划分，首都设为当前节点，如果不足，留在栈内与下一个子树合并。若没有子树了，加上当前节点留着往上回溯和上边的节点合并。到最后还剩不到B个，就和上一次划分的城市合并

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define LL long long int
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define Redge(u) for (int k = h[u]; k != -1; k = ed[k].nxt)
using namespace std;
const int maxn = 1005,maxm = 2005,INF = 1000000000;
inline int RD(){
    int out = 0,flag = 1; char c = getchar();
    while (c < 48 || c > 57) {if (c == '-') flag = -1; c = getchar();}
    while (c >= 48 && c <= 57) {out = (out << 1) + (out << 3) + c - '0'; c = getchar();}
    return out * flag;
}
int N,B,h[maxn],ne = 0;
struct EDGE{int to,nxt;}ed[maxm];
inline void build(int u,int v){
    ed[ne] = (EDGE){v,h[u]}; h[u] = ne++;
    ed[ne] = (EDGE){u,h[v]}; h[v] = ne++;
}
int st[maxn],fa[maxn],top = 0,cnt = 0,id[maxn],cap[maxn];
void dfs(int u){
    int to,last = top;
    Redge(u) if ((to = ed[k].to) != fa[u]){
        fa[to] = u;
        dfs(to);
        if (top - last >= B){
            cap[++cnt] = u;
            while (top > last) id[st[top--]] = cnt;
        }
    }
    st[++top] = u;
}
int main(){
    memset(h,-1,sizeof(h));
    N = RD(); B = RD();
    REP(i,N - 1) build(RD(),RD());
    dfs(1);
    while (top) id[st[top--]] = cnt;
    printf("%d\n%d",cnt,id[1]);
    for (int i = 2; i <= N; i++) printf(" %d",id[i]); printf("\n%d",cap[1]);
    for (int i = 2; i <= cnt; i++) printf(" %d",cap[i]);
    return 0;;
}