【BZOJ】【P4401】【块的计数】【题解】【???】

传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=4401

一开始写了一个O(约数个数*n)的……T了……

咦我记得n的约数个数好像不到O(sqrt(n))啊……(虽然肯定比logn大……)

忘记了TAT……求路过的同学告诉我n的约数个数究竟有多大……

枚举约数，统计子树size%B的个数

Code:

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
int n;
vector<int>G[maxn];
int fa[maxn],siz[maxn],lef[maxn];
int B;
void dfs(int u){
	siz[u]=1;	
	for(int i=0;i<G[u].size();i++){
		int v=G[u][i];
		if(v==fa[u])continue;
		fa[v]=u;
		dfs(v);
		siz[u]+=siz[v];
	}
}
int main(){
	scanf("%d",&n);
	for(int i=1;i<n;i++){
		int u,v;scanf("%d%d",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	dfs(1);
	int ans=0;
	static int cnt[maxn];
	for(int i=1;i<=n;i++)cnt[siz[i]]++;
	for(int i=1;i<=n;i++)if(n%i==0){
		int cot=0;
		for(int j=i;j<=n;j+=i)cot+=cnt[j];
		if(cot==n/i)ans++;
	}
	cout<<ans<<endl;
	return 0;
}