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);}