首先随便选一个根进行dfs得到size[x]表示以x为根节点的子树的大小。然后我们假设答案为t,需要判断t是否可行。首先显然需要t|n。显然每一个块有且仅有一个根,定义为这个块的最高点。然后我们发现一个点x是块的根的必要条件是t|size[x]!这个是显然的。然后我们统计有多少个size[x]被t整除,如果与n/t相同则合法,否则一定小于n/t,因此一定不合法。
时间复杂度O(N+N^0.5logN)。
AC代码如下:
#include<iostream>
#include<cstdio>
#define N 1000005
using namespace std;
int n,tot,fst[N],pnt[N<<1],nxt[N<<1],sz[N],sum[N];
void add(int aa,int bb){
pnt[++tot]=bb; nxt[tot]=fst[aa]; fst[aa]=tot;
}
void dfs(int x,int last){
int p; sz[x]=1;
for (p=fst[x]; p; p=nxt[p]){
int y=pnt[p];
if (y!=last){
dfs(y,x); sz[x]+=sz[y];
}
}
}
int read(){
int x=0; char ch=getchar();
while (ch<'0' || ch>'9') ch=getchar();
while (ch>='0' && ch<='9'){ x=x*10+ch-'0'; ch=getchar(); }
return x;
}
int main(){
n=read(); int i;
for (i=1; i<n; i++){
int x=read(),y=read(); add(x,y); add(y,x);
}
dfs(1,0); int ans=0; for (i=1; i<=n; i++) sum[sz[i]]++;
for (i=1; i<=n; i++) if (!(n%i)){
int tmp=0,j;
for (j=i; j<=n; j+=i) tmp+=sum[j];
if (tmp==n/i) ans++;
}
printf("%d\n",ans);
return 0;
}
by lych
2016.2.16