[bzoj1086]王室联邦

题目描述

“余”人国的国王想重新编制他的国家。他想把他的国家划分成若干个省，每个省都由他们王室联邦的一个成

员来管理。他的国家有n个城市，编号为1..n。一些城市之间有道路相连，任意两个不同的城市之间有且仅有一条

直接或间接的道路。为了防止管理太过分散，每个省至少要有B个城市，为了能有效的管理，每个省最多只有3B个

城市。每个省必须有一个省会，这个省会可以位于省内，也可以在该省外。但是该省的任意一个城市到达省会所经

过的道路上的城市（除了最后一个城市，即该省省会）都必须属于该省。一个城市可以作为多个省的省会。聪明的

你快帮帮这个国王吧！

greedy

由于我太弱了就去orz了hzwer的题解

无解情况是n<b。

接下来我们来弄一个贪心。

贪心方法见代码。

#include<cstdio>
#include<algorithm>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=1000+10;
int h[maxn],go[maxn*2],next[maxn*2],size[maxn],belong[maxn],cap[maxn],s[maxn];
int i,j,k,l,t,n,m,c,tot,top,cnt;
void add(int x,int y){
go[++tot]=y;
next[tot]=h[x];
h[x]=tot;
}
void dfs(int x,int y){
s[++top]=x;
int t=h[x];
while (t){
if (go[t]!=y){
dfs(go[t],x);
size[x]+=size[go[t]];
if (size[x]>=c){
size[x]=0;
cap[++cnt]=x;
while (s[top]!=x) belong[s[top--]]=cnt;
}
}
t=next[t];
}
size[x]++;
}
void dg(int x,int y,int z){
if (belong[x]) z=belong[x];else belong[x]=z;
int t=h[x];
while (t){
if (go[t]!=y) dg(go[t],x,z);
t=next[t];
}
}
int main(){
scanf("%d%d",&n,&c);
if (n<c){
printf("0\n");
return 0;
}
fo(i,1,n-1){
scanf("%d%d",&j,&k);
add(j,k);add(k,j);
}
dfs(1,0);
if (!cnt) cap[cnt=1]=1;
dg(1,0,cnt);
printf("%d\n",cnt);
fo(i,1,n) printf("%d%c",belong[i],(i==n)?'\n':' ');
fo(i,1,cnt) printf("%d%c",cap[i],(i==cnt)?'\n':' ');
}