BZOJ1086 SCOI2005王室联邦

最新推荐文章于 2018-05-22 14:41:33 发布

weixin_30713953

最新推荐文章于 2018-05-22 14:41:33 发布

阅读量111

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/nbwzyzngyl/p/8350543.html

本文介绍了一种树分块算法的实现思路与代码细节。通过深度优先搜索划分子树为多个区块，每个区块内的节点数量不超过预设阈值B。最终输出区块数量及各节点所属区块。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

想学树上莫队结果做了个树分块。

看完题后想到了菊花图的形状认为无解，结果仔细一瞧省会可以在外省尴尬

对于每一颗子树进行深搜，一旦遇到加在一起大小达到B则将它们并为一省，因为他子树搜完以后没有分出块的大小是小于B的，而且他自己当前剩下的也是小于B的，所以可以放心和。

最后剩下的点肯定也小于B所以与最后一个省一合即可。当前子树根节点留给上面也是因为这个。

By：大奕哥

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int N=1005;
 4 int bel[N],size[N],cap[N],head[N],num,cnt,top,s[N],n,b;
 5 struct node{
 6     int to,nex;
 7 }e[N<<1];
 8 void add(int x,int y)
 9 {
10     e[++cnt].nex=head[x];e[cnt].to=y;head[x]=cnt;
11 }
12 void dfs(int x,int fa)
13 {
14     s[++top]=x;
15     for(int i=head[x];i;i=e[i].nex)
16     {
17         int y=e[i].to;
18         if(y==fa)continue;
19         dfs(y,x);
20         if(size[x]+size[y]>=b){
21             size[x]=0;
22             cap[++num]=x;
23             while(s[top]!=x)bel[s[top--]]=num;
24         }
25         else size[x]+=size[y];
26     }
27     size[x]++;
28 }
29 int main()
30 {
31     scanf("%d%d",&n,&b);
32     if(n<b){
33         puts("0");return 0;
34     }
35     int x,y,w;
36     for(int i=1;i<n;++i)
37     {
38         scanf("%d%d",&x,&y);
39         add(x,y);add(y,x);
40     }
41     dfs(1,0);
42     while(top)bel[s[top--]]=num;
43     printf("%d\n",num);
44     for(int i=1;i<=n;++i)printf("%d ",bel[i]);
45     printf("\n");
46     for(int i=1;i<=num;++i)printf("%d ",cap[i]);
47     return 0;
48 }