树的计数

一个有n个结点的树,设它的结点分别为v1, v2, …, vn,已知第i个结点vi的度数为di,

问满足这样的条件的不同的树有多少棵。给定n,d1, d2, …, dn,
编程需要输出满足d(vi)=di的树的个数。

Input

第一行是一个正整数n,表示树有n个结点。
第二行有n个数,第i个数表示di,即树的第i个结点的度数。
其中1<=n<=150,输入数据保证满足条件的树不超过10^17个。

Output

输出满足条件的树有多少棵。

Sample Input

4
2 1 2 1
Sample Output

2

sol:本题利用Purfer序的一个性质,若树中一个结点的度为i,则它在Purfer序中出现的次数是i-1。

如上图,7个结点,1~7结点的度依次为2,3,1,3,1,1,1。各个点的度-1后依次为1,2,0,2,0,0,0。

purfer序为(2,2,1,4,4),即1号结点出现1次,2号结点出现2次,4号结点出现2次。

本题给出了结点个数及每个结点的度,我们就知道它在purfer序中出现的次数,从而算出满足条件的树。

如上图满足条件的树为c(5,1)*c(4,2)*c(2,2),即在purfer序的5个位置中,任选1个位置放1号结点,然后从剩下的4个位置中放2号结点,最后剩下2个位置放4号结点。

注意:实际求组合数的过程要分解质因数,不然爆long long。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#define ll long long
using namespace std;
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*=10;x+=ch-'0';ch=getchar();}
    return x*f;
}
int n,m,tot,cnt;
int d[155],num[155],pri[155];
ll s[25],ans=1;
bool jud(int x)
{
     for(int i=2;i<=sqrt(x);i++)
         if(x%i==0)return 0;
     return 1;
}
void getpri()
{
     for(int i=2;i<=150;i++)
         if(jud(i))pri[++cnt]=i;
}
void solve(ll x,int f)
//质因子分解,x中,第i个质因子pri[i]有num[i]个 
{
     for(int i=1;i<=cnt;i++)
     {
         if(x<=1)return;
         while(x%pri[i]==0)
         {num[i]+=f;x/=pri[i];}
     }
}
int main()
{
     s[1]=1;
     for(int i=2;i<=22;i++)
         s[i]=s[i-1]*i;
     getpri();
     n=read();
     if(n==1)
     {
         int x=read();
         if(!x)printf("1");
         else printf("0");
         return 0;
     }
     for(int i=1;i<=n;i++)
     {
         d[i]=read();
         if(!d[i]){printf("0");return 0;}
         d[i]--;//各个结点的度-1
         tot+=d[i];
     }
     if(tot!=n-2){printf("0");return 0;}
     //结果为n-2的阶乘/(d[1]的阶乘*d[2]的阶乘*...d[n]的阶乘)
  //多个组合数相乘约分得
     solve(s[n-2],1);//求解分子(n-2)! ,包含的质因子及指数
     for(int i=1;i<=n;i++)//求解分母,并用分子对应的质因子指数-分母的
         solve(s[d[i]],-1);
     for(int i=1;i<=cnt;i++)//枚举每个质因子
         while(num[i]--)//对于第i个质因子,计算pri[i]^num[i],并乘到乘积上去
             ans*=pri[i];
     printf("%lld",ans);
     return 0;
}
 
原文地址:https://www.cnblogs.com/cutepota/p/12070188.html