bzoj4401块的计数 dfs

题意：把一棵树分成几块，要求方案数，每个块内是一个联通块。
啊明显，只有那些x|n的数可能成为答案。
然后对于x进行判断，自己画图想想就能发现，只有x|sz[i]的sz[i]个数=n/x，即分成n/x块，每块x个数，才可行，大概解释一下就是，如果x|sz[i]，那么这个以i为根的子树肯定能分成sz[i]/x个块。
然后搞个桶来统计，每次不需要遍历所有sz，只需要在桶内log计算就可以了，这是一个小trick，一开始没想到，想了一会才想到。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
const int N=1e6+5;
typedef long long ll;
int head[N],next[N*2],go[N*2];
int p[N],tot,size[N],bz[N],n,q[N];
inline void add(int x,int y)
{
    go[++tot]=y;
    next[tot]=head[x];
    head[x]=tot;
}
inline void dfs(int x,int fa)
{
    size[x]=1;
    for(int i=head[x];i;i=next[i])
    {
        int v=go[i];
        if (v!=fa)
        {
            dfs(v,x);
            size[x]+=size[v];
            if (size[v]+1!=size[x])bz[x]=1;
        }
    }
}
int main()
{
    scanf("%d",&n);
    fo(i,1,n-1)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        add(x,y);
        add(y,x);   
    }
    int m=(int)sqrt(n),tot=0;   
    fo(i,1,m)
    {
        if (n%i==0)
        {
            p[++tot]=i;
            if (i*i!=n)
            p[++tot]=n/i;
        }
    }
    dfs(1,0);
    int ans=0;
    fo(i,1,n)q[size[i]]++;
    fo(i,1,n)
    if (n%i==0)
    {
        int tmp=0;
        for(int j=i;j<=n;j+=i)tmp+=q[j];
        if (tmp==n/i)ans++;
    }
    printf("%d\n",ans);
}