Description
Alice家里有一盏很大的吊灯。所谓吊灯,就是由很多个灯泡组成。只有一个灯泡是挂在天花板上的,剩下的灯泡都是挂在其他的灯泡上的。也就是说,整个吊灯实际上类似于一棵树。其中编号为1的灯泡是挂在天花板上的,剩下的灯泡都是挂在编号小于自己的灯泡上的。
现在,Alice想要办一场派对,她想改造一下这盏吊灯,将灯泡换成不同的颜色。她希望相同颜色的灯泡都是相连的,并且每一种颜色的灯泡个数都是相同的。
Alice希望你能告诉她,总共有哪些方案呢?
Alice是一个贪心的孩子,如果她发现方案不够多,或者太多了,就会很不高兴,于是她会尝试调整。对于编号为x(x≠1)的灯泡,如果原来是挂在编号为f[x]的灯泡上,那么Alice会把第x个灯泡挂到第 ( f[x] + 19940105 ) mod (x-1) + 1 个灯泡上。
由于九在古汉语中表示极大的数,于是,Alice决定只调整9次。对于原始状态和每一次调整过的状态,Alice希望你依次告诉她每种状态下有哪些方案。
对于100%的数据,n<=1.2*106。
Solution
我好像会n^2的树形dp?看一眼n的范围发现过不了
显然答案只会是n的约数,且k是答案当且仅当size为k的倍数的节点有至少n/k个
证明见PoPoQQQ的blog
Code
#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <queue>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define drp(i,st,ed) for (int i=st;i>=ed;--i)
#define fill(x,t) memset(x,t,sizeof(x))
const int N=2000000;
struct edge {int y,next;} e[N*2];
int r[N],d[N],queue[N],size[N];
int ls[N],fa[N],edCnt;
int read() {
int x=0,v=1; char ch=getchar();
for (;ch<'0'||ch>'9';v=(ch=='-')?(-1):(v),ch=getchar());
for (;ch<='9'&&ch>='0';x=x*10+ch-'0',ch=getchar());
return x*v;
}
void add_edge(int x,int y) {
e[++edCnt]=(edge) {y,ls[x]}; ls[x]=edCnt;
}
void top_sort(int n) {
std:: queue <int> que;
int head=1,tail=0;
rep(i,1,n) if (!d[i]) queue[++tail]=i;
for (;head<=tail;) {
int now=queue[head++];
for (int i=ls[now];i;i=e[i].next) {
if (!(--d[e[i].y])) {
que.push(e[i].y);
}
}
}
drp(i,n,1) {
size[i]=1;
for (int j=ls[i];j;j=e[j].next) {
size[i]+=size[e[j].y];
}
r[size[i]]++;
}
}
int main(void) {
int n=read();
rep(i,2,n) fa[i]=read();
rep(lxf,0,9) {
fill(ls,0); fill(r,0); edCnt=0;
rep(i,2,n) add_edge(fa[i],i);
top_sort(n);
printf("Case #%d:\n", lxf+1);
rep(i,1,n) if (n%i==0) {
int wjp=0;
for (int j=i;j<=n;j+=i) {
wjp+=r[j];
}
if (wjp>=n/i) {
printf("%d\n", i);
}
}
rep(i,2,n) fa[i]=(fa[i]+19940105)%(i-1)+1;
}
return 0;
}