题目描述
给出一棵n个点的树,求有多少个si使得整棵树可以分为n/si个连通块。
输入
第一行一个正整数N,表示这棵树的结点总数,接下来N-1行,每行两个数字X,Y表示编号为X的结点与编号为Y的结点相连。结点编号的范围为1-N且编号两两不同。
输出
一行一个整数Ans,表示所求的方案数。
样例输入
6
1 2
2 3
2 4
4 5
5 6
样例输出
3
题解
结论题
结论:k可行当且仅当子树大小是k的倍数的节点数有n/k个。
这结论好像挺显然的(只要你能想出来。。。)
然后就做完了。。。先预处理出每个节点的size,对size维护桶,枚举n的约数,再枚举它的倍数求和统计即可。
时间复杂度为 $O(\sum\limits_{k|n}\frac nk=\sum\limits_{k|n}k=n\log\log n)$
#include <cstdio> #include <cctype> #define N 1000010 int head[N] , to[N << 1] , next[N << 1] , cnt , si[N] , v[N]; inline void add(int x , int y) { to[++cnt] = y , next[cnt] = head[x] , head[x] = cnt; } void dfs(int x , int fa) { int i; si[x] = 1; for(i = head[x] ; i ; i = next[i]) if(to[i] != fa) dfs(to[i] , x) , si[x] += si[to[i]]; v[si[x]] ++ ; } inline char nc() { static char buf[100000] , *p1 , *p2; return p1 == p2 && (p2 = (p1 = buf) + fread(buf , 1 , 100000 , stdin) , p1 == p2) ? EOF : *p1 ++ ; } inline int read() { int ret = 0; char ch = nc(); while(!isdigit(ch)) ch = nc(); while(isdigit(ch)) ret = ((ret + (ret << 2)) << 1) + (ch ^ ‘0‘) , ch = nc(); return ret; } int main() { int n = read() , i , x , y , ans = 0; for(i = 1 ; i < n ; i ++ ) x = read() , y = read() , add(x , y) , add(y , x); dfs(1 , 0); for(i = 1 ; i <= n ; i ++ ) { if(!(n % i)) { for(x = 0 , y = i ; y <= n ; y += i) x += v[y]; if(x == n / i) ans ++ ; } } printf("%d\n" , ans); return 0; }