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;
    
    
    
    

     
     }