[SCOI2005]王室联邦

题目大意:
  给定一棵$n(n\leq1000)$个结点的无根树,对其结点进行分块。对于每个块至少在树中(不一定在块中)存在一个“关键点”,使得块中所有子结点和这个“关键点”构成一个联通块。并且每个块的大小$s$满足$b\leq s\leq3b$。求出任意一种分块的方案。

思路:
  贪心。
  任取一个结点作为根,从上往下DFS。用一个栈维护当前以访问完毕,还没有分配到块中的结点。对于当前以$u$为根的子树中还没有被分到块中的结点,枚举以$u$的每个子结点为根的子树,一旦其中几个未分配结点数之和大于等于$b$,就将它们分到一起,并将$u$设为关键点。将这些点从栈中删除。最后留在栈中的未分配的点和$u$一起留在栈中。
​  这样处理完整棵树后,肯定还会剩下一些点未分配。不难证明DFS时构造的块的大小$s$满足$b\leq s\leq2b$。最后剩下的点数$s$满足$s\leq b$。所以可以把最后剩下的那些点合并到任意与其相邻的块中,关键点为整棵树的根。

 1 #include<stack>
 2 #include<cstdio>
 3 #include<cctype>
 4 #include<vector>
 5 inline int getint() {
 6     register char ch;
 7     while(!isdigit(ch=getchar()));
 8     register int x=ch^'0';
 9     while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
10     return x;
11 }
12 const int N=1001;
13 int n,b,bel[N],cap[N];
14 std::stack<int> s;
15 std::vector<int> e[N];
16 inline void add_edge(const int &u,const int &v) {
17     e[u].push_back(v);
18     e[v].push_back(u);
19 }
20 void dfs(const int &x,const int &par) {
21     const int size=s.size();
22     for(unsigned i=0;i<e[x].size();i++) {
23         const int y=e[x][i];
24         if(y==par) continue;
25         dfs(y,x);
26         if((int)s.size()-size>=b) {
27             cap[++cap[0]]=x;
28             while((int)s.size()>size) {
29                 bel[s.top()]=cap[0];
30                 s.pop();
31             }
32         }
33     }
34     s.push(x);
35 }
36 int main() {
37     n=getint(),b=getint();
38     for(register int i=1;i<n;i++) {
39         add_edge(getint(),getint());
40     }
41     dfs(1,0);
42     while(!s.empty()) {
43         bel[s.top()]=cap[0];
44         s.pop();
45     }
46     printf("%d\n",cap[0]);
47     for(register int i=1;i<=n;i++) {
48         printf("%d%c",bel[i]," \n"[i==n]);
49     }
50     for(register int i=1;i<=cap[0];i++) {
51         printf("%d%c",cap[i]," \n"[i==cap[0]]);
52     }
53     return 0;
54 }

 

posted @ 2018-02-18 22:36  skylee03  阅读(100)  评论(0编辑  收藏  举报