植树
题目描述
Ysuperman 有一棵 nn 个节点的无根树 TT。如果你不知道树是什么,TA 很乐意告诉你,树是一个没有环的无向联通图。
既然树是无根的,那就没有办法种植。Ysuperman 研究了很久的园艺,发现一个节点如果可以成为根,它必须十分平衡,这意味着以它为根时,与它直接相连的节点,他们的子树大小都相同。
你作为幼儿园信息组一把手,Ysuperman 给你一棵树,你能在 1s1s 内找到所有可能成为根的节点吗?
分析
大致思路为求出每个节点作为根节点时,是否满足条件。每个子树的节点个数时时必须要知道的。我们可以任意选取一个点为根节点,来计算每个子树大小(d[x])
。以下以选取1
号点为例。
void dfs(int u, int fa)
{
d[u] = 1;
for(int i = h[u]; i != -1; i = ne[i])
{
int j = e[i];
if(j == fa) continue;
dfs(j, u);
d[u] += d[j];
}
if(u != 1 && num && num != n - d[u]) root[u] = 0;
}
在求以u
为根节点的子树大小的过程中,我们可以顺便并比较不同子树大小是否相同,可以设定num
,用来记录其中一个子树的大小。不想打则x
不能成为根节点
if(!num) num = d[j];
else
{
if(d[j] != num) root[u] = 0;
}
在上述比较中,我们只能比较**x
结点以下的子树是否相同,所以还需要再比较一下其上方的部分**,根节点1
或者不存在子树的,也需要再进一步筛出。
if(u != 1 && num && num != n - d[u]) root[u] = 0;
完整代码
#include<bits/stdc++.h>
using namespace std;
const int N = 1e6 + 10, M = 2 * N;
int n, m, k, t;
int h[N], e[M], ne[M], idx;
int d[N], root[N];
void add(int a, int b)
{
e[idx] = b, ne[idx] = h[a], h[a] = idx ++;
}
void dfs(int u, int fa)
{
int num = 0;
d[u] = 1;
for(int i = h[u]; i != -1; i = ne[i])
{
int j = e[i];
if(j == fa) continue;
dfs(j, u);
d[u] += d[j];
if(!num) num = d[j];
else
{
if(d[j] != num) root[u] = 0;
}
}
if(u != 1 && num && num != n - d[u]) root[u] = 0;
}
int main()
{
memset(root, 1, sizeof root);
memset(h, -1, sizeof h);
cin >> n;
for(int i = 1; i < n; i ++)
{
int a, b;
cin >> a >> b;
add(a, b), add(b, a);
}
dfs(1,0 );
int cnt = 0;
for(int i = 1; i <= n; i ++)
if(root[i]) cout << i << " ";
puts("");
for(int i = 1; i <= n; i ++) cout <<d[i] << "--";
return 0;
}