树上摩托 sherco.cpp

【一句话题意】给定一棵树，问剪成几棵节点数相同的子树的方案数。节点数n<=1e6
【分析】易证当每棵子树的节点数相同时，划分方案是唯一的。问题在于确定点数并进行判定。本来是一道愉悦身心的好题，结果n是1e6。考场上是先根号n筛n的因数，再用dfs判定，结果。。。顺利超时。
然后，正解是把dfs转化为O(n)的线性扫描，原因是常数小。。。。。
再次投诉出题人，竟然卡dfs。
【code】

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+1000;
struct Edge{
	int v,nxt;
}edge[maxn<<1];
int head[maxn],tot;
int n;
int cnt[maxn];
inline void read(int &x){
	x=0;char tmp=getchar();
	while(tmp<'0'||tmp>'9') tmp=getchar();
	while(tmp>='0'&&tmp<='9') x=(x<<1)+(x<<3)+tmp-'0',tmp=getchar();
}
inline void add_edge(int u,int v){
	edge[tot].v=v;
	edge[tot].nxt=head[u];
	head[u]=tot++;
}
void dfs(int u,int fa){
	cnt[u]=1;
	for(int i=head[u];i!=-1;i=edge[i].nxt){
		int v=edge[i].v;
		if(v==fa) continue;
		dfs(v,u);
		cnt[u]+=cnt[v];
	}
}
int main(){
	memset(head,-1,sizeof(head));
	cin>>n;
	for(int i=1;i<n;i++){
		int u,v;read(u),read(v);
		add_edge(u,v);
		add_edge(v,u);
	}
	dfs(1,-1);
	int res,p,sqr=sqrt(n),ans=0;
	for(int i=1;i<=sqr;i++){
		p=i;
		if(n%p==0){
			res=0;
			for(int j=1;j<=n;j++)
				if(cnt[j]%p==0)res++;
			ans=ans+(res==n/p);
			if(i*i!=n){
				p=n/i,res=0;
				for(int j=1;j<=n;j++)
					if(cnt[j]%p==0)res++;
				ans=ans+(res==n/p);
			}
		}
	}
	cout<<ans<<endl;
	return 0;
}