BZOJ4401 块的计数

#include<bits/stdc++.h>
using namespace std; 
const int maxn=1000004;
int sz[maxn],ans[maxn],tot=0,cnt=0,head[maxn],n;
struct node{
    int to,next;
}e[maxn<<1];
inline void add(int u,int v){
    e[cnt].to=v;e[cnt].next=head[u];head[u]=cnt++;
}
inline void dfs(int u,int fa){
    sz[u]=1;
    for(int i=head[u];i!=-1;i=e[i].next){
        int v=e[i].to;
        if(v!=fa){
            dfs(v,u);
            sz[u]+=sz[v];
        }
    }
    ans[sz[u]]++;
}
int main(){
    memset(head,-1,sizeof(head));
    scanf("%d",&n);
    for(int i=1;i<n;i++){
        int x,y;scanf("%d%d",&x,&y);
        add(x,y);add(y,x);
    }
    dfs(1,0);
    for(int i=1;i<=n;i++){
        int x=0;
        for(int j=i;j<=n;j+=i) x+=ans[j];
        if(x*i==n) tot++;
    }
    cout<<tot;
}