来自学长的快乐AK题——Day5 C

C——蜜汁做法

Description

一棵树删去任意条边，可以得到若干棵新树。
若使让每棵新树的节点数相同。问有多少种方法分割这棵树。
设给出的树的结点数为n，并给出n-1条边

输入

第一行一个正整数N，表示这棵树的结点总数。
接下来N-1行，每行两个数字X，Y表示编号为X的结点与编号为Y的结点相连。结点编号的范围为[1,N]。

输出

一个整数，表示方案数。注意，不砍去任何一条边也算作一种方案。

数据规模

对于40%的数据，N ≤ 15
对于60%的数据，N ≤ 10^5
对于100%的数据，N ≤ 10^6
数据规模非常大，请使用高效的读入方式。

思路

（扣自教练题解）

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
const int N=5*1e6;
int n,size[N],suma,sumb,a,b,ans;
int he[N],ne[N],to[N],cnt;
bool vis[N];
void add(int u,int v)
{
	cnt++;
	ne[cnt]=he[u];
	he[u]=cnt;
	to[cnt]=v;
}
int read()
{
	char c=getchar();
	while(!isalnum(c)) c=getchar();
	int x=0;
	while(isalnum(c))
	x=(x<<3)+(x<<1)+c-48,c=getchar();
	return x;
}
void dfs(int u)//预处理每个节点的子树的节点数size
{
	vis[u]=1;
	for(int i=he[u];i;i=ne[i])
	{
		int v=to[i];
		if(!vis[v])
		{
			dfs(v);
			size[u]+=size[v];
		}
	}
	size[u]++;
}
int main()
{
	scanf("%d",&n);
	for(int i=1,u,v;i<n;i++)
	{
		u=read(),v=read();
		add(u,v),add(v,u);
	}
	dfs(1);
	for(int i=1;i*i<=n;i++)//枚举n的约数
	{
		if(n%i) continue;
		a=i,b=n/i,suma=sumb=0;
		for(int j=1;j<=n;j++)
		{
			if(size[j]%a==0) suma++;
			if(size[j]%b==0) sumb++;
		}
		if(suma==n/a)
		{
			ans++;
			if(a==b) ans--;//考虑完全平方数
		}
		if(sumb==n/b) ans++;
	}
	if(n==1) ans=1;//特判呜呜呜
	printf("%d",ans);
	return 0;
}