bzoj3004 吊灯 乱搞+结论题

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;
}