1086: [SCOI2005]王室联邦

一开始做的时候脑残了，以为会出现很多无解的情况，于是加了一堆特判，果断WA。

后来发现，明明只有n<B的时候才无解啊。

然后发现题意理解错了。。。。

原来多颗子树可以由根连接成省，而根可以在省外。。。。。。。

于是我们有个贪心的方法。

当前节点的子树集的大小一旦大于等于B，直接分成一个省，当然我们还可以证明这个省的城市数<=2B。

最后还有一些<B的省，直接塞到别的省就好了。

#include<iostream>#include<cstdio>#include<cstring>using namespace std;struct Edge{int to,next;}e[2010];int head[1005],cnt,k;void ins(int u,int v){cnt++;e[cnt].to=v;e[cnt].next=head[u];head[u]=cnt;}int B,sz[1005],q[1005],top,cap[1005],pro,belong[1005];void dfs(int u,int fa){q[++top]=u;for(int i=head[u];i;i=e[i].next){int v=e[i].to;if(v==fa)continue;dfs(v,u);if(sz[u]+sz[v]>=B){sz[u]=0;cap[++pro]=u;while(q[top]!=u)belong[q[top--]]=pro;}else sz[u]+=sz[v];}sz[u]++;}void set(int u,int fa,int anc){if(belong[u])anc=belong[u];else belong[u]=anc;for(int i=head[u];i;i=e[i].next){int v=e[i].to;if(v==fa)continue;set(v,u,anc);}}int main(){int n;scanf("%d%d",&n,&B);if(n<B){printf("0");return 0;}int u,v;for(int i=1;i<n;i++){scanf("%d%d",&u,&v);ins(u,v);ins(v,u);}dfs(1,0);if(!pro)cap[++pro]=1;set(1,0,pro);printf("%d\n",pro);for(int i=1;i<=n;i++)printf("%d ",belong[i]);printf("\n");for(int i=1;i<=pro;i++)printf("%d ",cap[i]); return 0;}