bzoj4401 块的计数 dfs

       首先随便选一个根进行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