1086: [SCOI2005]王室联邦

Time Limit: 10 Sec Memory Limit: 162 MBSec Special Judge
Submit: 1554 Solved: 933
[Submit][Status][Discuss]

Description

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

Input

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

Output

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

Sample Input

8 2
1 2
2 3
1 8
8 7
8 6
4 6
6 5

Sample Output

3
2 1 1 3 3 3 3 2
2 1 8

HINT

Source

大神们都一句话带过。

在网上扒了老长时间，才领悟。

整理一下，发给大家。

先说一个问题：

如果在当前节点，其有两个儿子。

在左儿子的子树中搜得的节点数是a-1的话，这时候我们去搜右儿子，不妨假设右儿子为根的子树是一条长为10000的链。

那么显然最下面的叶节点跟左儿子代表的子树是在同一块中的，这显然碎成渣了。

所以每一次搜到一个节点，我们都要把当前的栈的指针当做该节点的相对栈底，这样的话就能避免这个问题。

首先以任意节点为根DFS，DFS的任务是解决以当前节点为根的子树（不包括当前节点）中的节点的归属，并汇报不知道去向何方者。

具体做法是：

对于每个正在处理的节点v，定义一个等待序列，扫一遍它的孩子。在扫的时候，假设当前for循环正在处理的是孩子u，DFS(u)，然后把传回来的等待序列加到v的等待序列中。如果v的等待序列节点数超过了B，那么就让等待队列中的节点组成一个省，省会是v，但v不划入那个省中。最后，for循环结束，把v加入到等待序列中，return。在主函数中接收DFS的返回值，若等待序列为空，皆大欢喜；但事实上，等待序列不可能为空。

那么等待序列中的节点数一定不超过B + 1，怎么办？答案就是放在上一个被划分出的省中去，那么上一个被划分出的省一定<=2B - 1，而现在最后无家可归的节点数一定<= B + 1，所以放在上一个被划分出的省中去节点数一定不超过3B！而且最后被划分出来的省一定和最后无家可归的节点中的某一个相邻！那么考虑什么时候会出现无解呢？答案就是我们找不到“上一个被划分出的省”

#include<cstdio>

using namespace std;

const int N=1e4+;

struct node{

    int v,next;

}e[N<<];

int n,B,top,tot,cnt,head[N],belong[N],root[N],stack[N];

void add(int x,int y){

    e[++tot].v=y;e[tot].next=head[x];head[x]=tot;

    e[++tot].v=x;e[tot].next=head[y];head[y]=tot;

}

void dfs(int x,int fa){

    int bottom=top;

    for(int i=head[x];i;i=e[i].next){

        if(e[i].v!=fa){

            dfs(e[i].v,x);

            if(top-bottom>=B){

                root[++cnt]=x;

                for(;top!=bottom;top--) belong[stack[top]]=cnt;

            }

        }

    }

    stack[++top]=x;

}

int main(){

    scanf("%d%d",&n,&B);

    for(int i=,x,y;i<n;i++) scanf("%d%d",&x,&y),add(x,y);

    dfs(,);

    for(;top;top--) belong[stack[top]]=cnt;

    printf("%d\n",cnt);

    for(int i=;i<=n;i++) printf("%d ",belong[i]);putchar('\n');

    for(int i=;i<=cnt;i++) printf("%d ",root[i]);putchar('\n');

    return ;

}