bzoj 4401: 块的计数 水题

水题，想麻烦了一点点

我的思路：显然只有sz|n时才有可能合法，一个数的约数个数时O(logn)的，所以可以暴力dp

题解：

其实，当且仅当一棵子树的sz[z]是块的大小的倍数，这个子树的根可以成为一个块的根，所以枚举一下答案暴力就行。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
 
#define ll long long
#define inf 1e9
#define eps 1e-8
#define md
#define N 1000010 
using namespace std;
struct yts { int x,t,ne;} e[2*N];
int v[N],sz[N],c[N];
int num;
 
void put(int x,int y)
{
num++; e[num].x=x; e[num].t=y;
e[num].ne=v[x]; v[x]=num;
}
 
void dfs(int x,int fa)
{
sz[x]=1;
for (int i=v[x];i;i=e[i].ne)
{
int y=e[i].t;
if (y!=fa)
{
dfs(y,x);
sz[x]+=sz[y];
}
}
c[sz[x]]++;
}
 
int main()
{
int n;
scanf("%d",&n);
for (int i=1;i<n;i++)
{
int x,y;
scanf("%d%d",&x,&y);
put(x,y); put(y,x);
}
dfs(1,0);
int ans=0;
for (int i=1;i<=n;i++)
  if (n%i==0)
  {
  int tot=0;
  for (int j=i;j<=n;j+=i) tot+=c[j];
if (tot==n/i) ans++;
  }
printf("%d\n",ans);
return 0;
}