【普吕弗编码 + 组合数学】bzoj 1211: [HNOI2004]树的计数

本文详细介绍了HNOI2004树的计数问题,利用普吕弗编码原理,阐述了如何通过给定点的度数计算可能的树的数量。并提供了AC代码实现,包括初始化质数、求解过程和最终输出。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

1211: [HNOI2004]树的计数

Description

一个有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


思路

根据普吕弗编码(关于Pruper Sequence
https://blog.youkuaiyun.com/baiyifeifei/article/details/84065899 ),我们已知,任何一棵树,都唯一对应一段长度为n-2的序列,序列中每个点出现的次数都等于其在树中的度数-1,因此假设有n个点,我们可以得到答案
ans=An−2n−2∏i=1n(du[i]−1)! ans=\frac{A_{n-2}^{n-2}}{\prod_{i=1}^{n}(du[i]-1)!} ans=i=1n(du[i]1)!An2n2
值得注意的是当只有一个节点时,需要特判,假如这个点度数不为0则无法构成树,而点大于1时,不能有点的度数为0,且必须出现的数的次数之和必须等于n-2


AC代码

/**************************************************************
    Problem: 1211
    User: FlyWhite
    Language: C++
    Result: Accepted
    Time:0 ms
    Memory:1292 kb
****************************************************************/
 
#include<cstdio>
#include<iostream>
#include<cmath>
#include<string>
#include<cstring>
using namespace std;
typedef long long LL;
const int size=160;
int du[size];
int tim[size];
bool prime[size];
int p[size],tot;
void init_prime(int n)
{
    tot=0;
    fill(prime,prime+size,true);
    for(int i=2;i<=n;i++)
    {
        if(prime[i] ) p[tot++]=i;
        for(int j=0;j<tot&&i*p[j]<n;j++)
        {
            prime[i*p[j]]=false;
            if(i%p[j]==0) break;
        }
    }
}
void solve(int num,int op)
{
    for(int i=2;i<=num;i++)
    {
        int w=i;
        for(int j=0;j<tot;j++)
        {
            if(w%p[j]==0)
            while(w%p[j]==0)
            {
                w/=p[j];
                tim[j]+=op;
            }
        }
    }
}
int main()
{
    int n;
    scanf("%d",&n);
    int sum=0;
    memset(tim,0,sizeof(tim));
    init_prime(n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&du[i]);
        if(n==1) continue;
        if(!du[i])
        {
            printf("0\n");
            return 0;
        }
        du[i]--;
        sum+=du[i];
    }
    if(n==1) {
        if(!du[1]) printf("1\n");
        else printf("0\n");
        return 0;
    }
    if(sum!=n-2) {
        printf("0");
        return 0;
    }
    solve(n-2,1);
    long long ans=1;
    for(int i=1;i<=n;i++)
    {
        if(du[i]>1)solve(du[i],-1);
    }
    for(int i=0;i<tot;i++)
    {
        for(int j=1;j<=tim[i];j++)
        {
            ans*=p[i];
        }
    }
    printf("%lld",ans);
}

转载于:https://www.cnblogs.com/fly-white/p/10092689.html

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值