题目
Alice家里有一盏很大的吊灯。所谓吊灯,就是由很多个灯泡组成。只有一个灯泡是挂在天花板上的,剩下的灯泡都是挂在其他的灯泡上的。也就是说,整个吊灯实际上类似于[b]一棵树[/b]。其中编号为 1 的灯泡是挂在天花板上的,剩下的灯泡都是挂在编号小于自己的灯泡上的。
现在,Alice想要办一场派对,她想改造一下这盏吊灯,将灯泡换成不同的颜色。她希望相同颜色的灯泡都是相连的,并且每一种颜色的灯泡个数都是相同的。
Alice希望你能告诉她,总共有哪些方案呢?
Alice是一个贪心的孩子,如果她发现方案不够多,或者太多了,就会很不高兴,于是她会尝试调整。对于编号为[b]x(x≠1)[/b]的灯泡,如果原来是挂在编号为[b]f[x][/b]的灯泡上,那么Alice会把第x个灯泡挂到[b]第 ( f[x] + 19940105 ) mod (x-1) + 1 个[/b]灯泡上。
由于九在古汉语中表示极大的数,于是,[b][color=#FF0000]Alice决定只调整9次[/color][/b]。对于原始状态和每一次调整过的状态,Alice希望你依次告诉她每种状态下有哪些方案。
输入格式
第一行一个整数n,表示灯泡的数量。
接下来一行,有n-1个整数Ui,第i个数字表示第i+1个灯泡挂在了Ui个的下面。保证编号为1的灯泡是挂在天花板上的。数字之间用逗号(西文标点)" , "隔开且最后一个数字后面没有逗号。
输出格式
对于10种状态下的方案,需要按照顺序依次输出。
对于每一种状态,需要先输出单独的一行,表示状态编号,如样例所示。
之后若干行,每行1个整数,表示划分方案中每种颜色的灯泡个数。
[b]按升序输出。[/b]
输入样例
6
1,2,3,4,5
输出样例
Case #1:
1
2
3
6
Case #2:
1
2
6
Case #3:
1
3
6
Case #4:
1
3
6
Case #5:
1
3
6
Case #6:
1
2
6
Case #7:
1
2
3
6
Case #8:
1
6
Case #9:
1
2
6
Case #10:
1
3
6
提示
对于20%的数据,n<=3*10^3。
对于40%的数据,n<=5*10^4。
对于50%的数据,n<=1*10^5。
对于60%的数据,n<=3*10^5。
对于70%的数据,n<=7*10^5。
对于100%的数据,n<=1.2*10^6。
题解
题意是要把树分成等份的块,有哪些分法合法
枚举块的大小x
首先一定是n的因子,再者一定存在(frac{n}{x})个节点,其子树大小是x的倍数
证明:
对于大小是x的倍数的子树,一定能从根节点开始分出一个大小为x的块
如果其子孙中存在大小同样是x的倍数,那么无视该子孙,剩余的节点连续且数量同样是x的倍数,同样可以分出一个大小为x的块
故一个大小为x的倍数的子树可以在不影响其它大小为x的倍数的子树的情况下分出一个大小为x的块
由因为有(frac{n}{x})个这样的子树,故可以将n分为(frac{n}{x})个大小为x的块
证毕
由于每次儿子一定比父亲大,没必要dfs构树,可以直接逆向枚举构树就可以了
#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define LL long long int
#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)
#define REP(i,n) for (int i = 1; i <= (n); i++)
#define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<' '; puts("");
using namespace std;
const int maxn = 1200005,maxm = 100005,INF = 1000000000;
inline int read(){
int out = 0,flag = 1; char c = getchar();
while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}
while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}
return out * flag;
}
int f[maxn],siz[maxn],bac[maxn],n;
inline bool check(int x){
int tot = 0;
for (int i = x; i <= n; i += x) tot += bac[i];
return tot >= n / x;
}
int main(){
n = read();
for (int i = 2; i <= n; i++) f[i] = read();
for (int T = 1; T <= 10; T++){
printf("Case #%d:
",T);
for (int i = 1; i <= n; i++) bac[i] = 0,siz[i] = 1;
for (int i = n; i; i--) siz[f[i]] += siz[i],bac[siz[i]]++;
for (int i = 1; i <= n; i++){
if (n % i == 0 && check(i))
printf("%d
",i);
}
if (T < 10)
for (int i = 2; i <= n; i++) f[i] = (f[i] + 19940105) % (i - 1) + 1;
}
return 0;
}