首页 > 代码库 > bzoj3004 吊灯
bzoj3004 吊灯
3004: 吊灯
Time Limit: 10 Sec Memory Limit: 128 MBSubmit: 52 Solved: 32
[
id=3004" style="color:blue; text-decoration:none">Submit][Status][Discuss]
Description
所谓吊灯,就是由非常多个灯泡组成。仅仅有一个灯泡是挂在天花板上的,剩下的灯泡都是挂在其它的灯泡上的。也就是说。整个吊灯实际上类似于一棵树。当中编号为1的灯泡是挂在天花板上的,剩下的灯泡都是挂在编号小于自己的灯泡上的。
对于原始状态和每一次调整过的状态。Alice希望你依次告诉她每种状态下有哪些方案。
Input
数字之间用逗号‘,’隔开且最后一个数字后面没有逗号。
Output
Sample Input
1,2,3,4,5
Sample Output
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
HINT
对于100%的数据,n<=1.2*106。
有一个结论:k满足条件当且仅当存在n/k个节点满足以该节点为根的子树大小是k的倍数。
证明:如果有n/k个大小为k的连通块。那对于每一个连通块中深度最小的节点。相应子树大小一定是k的倍数,必要性得证。如果存在n/k个大小为k的倍数的子树,那我们DFS一遍。每次遇到一个满足条件的节点,就将它砍下来,终于肯定会得到n/k个连通块,充分性得证。
然后求出每个点字数大小,f[i]表示大小为i的子树个数对于,对于每个约数扫一遍f数组。推断是否可行
#include<iostream> #include<cstdio> #include<cstdlib> #include<cmath> #include<cstring> #include<algorithm> #define F(i,j,n) for(int i=j;i<=n;i++) #define D(i,j,n) for(int i=j;i>=n;i--) #define ll long long #define maxn 1200005 using namespace std; int n,cnt; int fa[maxn],f[maxn],p[maxn],size[maxn]; inline int read() { int x=0,f=1;char ch=getchar(); while (ch<'0'||ch>'9'){if (ch=='-') f=-1;ch=getchar();} while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } int main() { n=read(); for(int i=1;i*i<=n;i++) if (n%i==0) { p[++cnt]=i; if (i*i!=n) p[++cnt]=n/i; } sort(p+1,p+cnt+1); F(t,1,10) { printf("Case #%d:\n",t); memset(size,0,sizeof(size)); memset(f,0,sizeof(f)); F(i,2,n) fa[i]=t==1?read():(fa[i]+19940105)%(i-1)+1; D(i,n,1) size[fa[i]]+=(++size[i]); F(i,1,n) f[size[i]]++; F(i,1,cnt) { int tmp=0; F(j,1,n/p[i]) tmp+=(f[p[i]*j]); if (tmp==n/p[i]) printf("%d\n",p[i]); } } return 0; }
bzoj3004 吊灯