首页 > 代码库 > BZOJ 1005 明明的烦恼 Prufer序列+组合数学+高精度
BZOJ 1005 明明的烦恼 Prufer序列+组合数学+高精度
题目大意:给定一棵n个节点的树的节点的度数,其中一些度数无限制,求可以生成多少种树
Prufer序列
把一棵树进行以下操作:
1.找到编号最小的叶节点,删除这个节点,然后与这个叶节点相连的点计入序列
2.反复进行1,直到这棵树只剩下两个节点时,退出
比如说这个图(来自度受百科)
最小叶节点为2,删除2,将3计入序列
最小叶节点为4,删除4,将5计入序列
最小叶节点为5,删除5,将1计入序列
最小叶节点为1,删除1,将3计入序列
图中只剩下两个节点,退出
于是得到这棵树的Prufer序列为{3,5,1,3}
这样可以得到一个长度为n-2的序列。很容易证明,树和Prufer序列是一一对应的
Prufer序列显然满足一个性质:一个点若度数为d,则一定在Prufer序列中出现了d-1次
于是这就变成了一个排列组合的问题了
令每个已知度数的节点的度数为di,有n个节点,m个节点未知度数,left=(n-2)-(d1-1)-(d2-1)-...-(dk-1)
已知度数的节点可能的组合方式如下
(n-2)!/(d1-1)!/(d2-1)!/.../(dk-1)!/left!
剩余left个位置由未知度数的节点随意填补,方案数为m^left
于是最后有
ans=(n-2)!/(d1-1)!/(d2-1)!/.../(dk-1)!/left! * m^left
答案很显然有高精度,为了避免高精度除法我们可以对每个阶乘暴力分解质因数,对指数进行加减操作即可
#include<cstdio> #include<cstring> #include<iostream> #include<algorithm> #define M 1100 using namespace std; typedef long long ll; struct abcd{ ll xx[400]; int cnt; abcd(int x=0) { memset(xx,0,sizeof xx); xx[1]=x; cnt=1; } ll& operator [] (int x) { return xx[x]; } }ans(1); abcd operator *= (abcd &x,abcd &y) { int i,j; abcd z; for(i=1;i<=x.cnt;i++) for(j=1;j<=y.cnt;j++) z[i+j-1]+=x[i]*y[j],z[i+j]+=z[i+j-1]/100000000,z[i+j-1]%=100000000; z.cnt=x.cnt+y.cnt; if(!z[z.cnt]) --z.cnt; x=z; } ostream& operator << (ostream& os,abcd &x) { int i; printf("%lld",x[x.cnt]); for(i=x.cnt-1;i;i--) printf("%08lld",x[i]); return os; } int n,m,remain,cnt[M],stack[M],top; void Decomposition(int x,int y) { int i; for(i=2;i*i<=x;i++) while(x%i==0) cnt[i]+=y,x/=i; if(x^1) cnt[x]+=y; } void Quick_Power(int i,int y) { abcd x(i); while(y) { if(y&1)ans*=x; x*=x; y>>=1; } } int main() { int i,x; cin>>n;remain=n-2; for(i=1;i<=n;i++) { scanf("%d",&x); if(x==-1) ++m; else if(x>1) stack[++top]=x-1,remain-=x-1; } for(i=2;i<=n-2;i++) Decomposition(i,1); while(top) { for(i=2;i<=stack[top];i++) Decomposition(i,-1); stack[top--]=0; } for(i=2;i<=remain;i++) Decomposition(i,-1); Decomposition(m,remain); for(i=1;i<=n;i++) if(cnt[i]) Quick_Power(i,cnt[i]); cout<<ans<<endl; }
BZOJ 1005 明明的烦恼 Prufer序列+组合数学+高精度
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。