bzoj4401 块的计数

　　首先确定一个根后dfs一遍，统计出每个子树的点数个数，然后枚举n的因子x，统计多少子树的点数个数是当前枚举因子x的倍数，如果有n/x个子树满足条件，则答案+1

　　代码

　　

#include<cstdio>
#include<algorithm>
#include<set>
#define fi first
#define sc second
#define mp make_pair
#define N 2000100
using namespace std;
int p[N],pre[N],tt[N],s[N],cnt[N];
int n,i,a,b,j,ans,dp,sum;
void link(int x,int y)
{
    dp++;pre[dp]=p[x];p[x]=dp;tt[dp]=y;
}
void dfs(int x,int fa)
{
    int i=p[x];
    s[x]=1;
    while (i)
    {
        if (tt[i]!=fa)
        {
            dfs(tt[i],x);
            s[x]+=s[tt[i]];
        }
        i=pre[i];
    }
}
int main()
{
    scanf("%d",&n);
    for (i=1;i<n;i++)
    {
        scanf("%d%d",&a,&b); 
        link(a,b);
        link(b,a); 
    }
    dfs(1,0);
    for (i=1;i<=n;i++)
    cnt[s[i]]++;
    for (i=1;i<=n;i++)
    if (n%i==0)
    {
        sum=0;
        for (j=i;j<=n;j+=i)
        sum+=cnt[j];
        if (sum==n/i) ans++;
    }
    printf("%d
",ans);
}