【Prufer编码+组合】BZOJ1005(HNOI2008)[明明的烦恼]题解

题目概述

有 $n$ 个点，每个点的度数为 $d_i$ （若 $d_i=-1$ 则度数没有限制），求满足条件的树的个数。

解题报告

根本不会做好吗！首先介绍一下Prufer编码：

• 树 $\to$ Prufer编码
  
  • 找到编号最小的度为 $1$ 的节点，删除该点及连接该点的边，将与该点相邻的点加入数列。
  • 重复删除/加入，直到剩下两个节点，最后的数列就是Prufer编码（长度为 $n-2$ ）。
• Prufer编码 $\to$ 树
  
  • 令 $x$ 是**目前**Prufer编码中没出现的最小编号，将 $x$ 与数列的第一项连边，删除第一项。
  • 重复连边/删除，直到序列为空，此时仍有两个节点没有被选出过，进行连边。

重要性质：一棵树与一个Prufer编码唯一对应。

所以这道题转化为求Prufer编码的个数。

---

还有一个比较显然的性质，如果一个节点的度数为 $D$ ，那么该节点将会在Prufer编码中出现 $D-1$ 次。

我们先无视没有限制的点，令 $tot=\sum\{d_i-1|d_i\not =-1\}$ 。首先，我们要为这些点腾出 $tot$ 个位置，方案数为 $n-2\choose tot$ ，然后要为第一个点腾出 $d_1-1$ 个位置，方案数为 $tot\choose d_1-1$ ，同理要为第二个点腾出 $d_2-1$ 个位置，方案数为 $tot-(d_1-1)\choose d_2-1$ ……

展开成阶乘的形式然后约分，得到答案为 $(n-2)!\over (n-2-tot)!(d_1-1)!\cdots(d_n-1)!$ ，挖一下素数然后套阶乘素数公式就行了。

最后就剩下不受限制的点了，其实也很容易分析。因为不受限制，所以剩下的位置可以乱填，令 $m=$ 不受限制的点的个数，那么方案数为 $m^{n-2-tot}$ 。

---

从最后可以看出，若要求 $n$ 个节点树的个数，则应该是 $n^{n-2}$ 个。这其实就是Cayley定理。

示例程序

像我这么懒的人，高精度肯定是copy的……

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1000,maxl=1000,Base=10000;

int n,d[maxn+5],m,tot;
int p[maxn+5];bool pri[maxn+5];
struct bignum
{
    int num[maxl+5];bignum(){};
    void operator = (int x) {memset(num,0,sizeof(num));do num[++num[0]]=x%Base,x/=Base; while (x);}
    bignum(int x) {*this=x;}
    void print() {printf("%d",num[num[0]]);for (int i=num[0]-1;i>=1;i--) printf("%04d",num[i]);}
    void Update() {while (num[num[0]+1]) num[0]++;while (num[0]>1&&!num[num[0]]) num[0]--;}
};
bignum ans,c;
inline bignum operator * (const bignum &a,const bignum &b)
{
    memset(c.num,0,sizeof(c.num));c.num[0]=a.num[0]+b.num[0]-1;
    for (int i=1;i<=a.num[0];i++)
    for (int j=1;j<=b.num[0];j++)
    {
        c.num[i+j-1]+=a.num[i]*b.num[j];
        c.num[i+j]+=c.num[i+j-1]/Base;
        c.num[i+j-1]%=Base;
    }
    return c.Update(),c;
}

void Make()
{
    pri[0]=pri[1]=true;
    for (int i=2;i<=maxn;i++)
    {
        if (!pri[i]) p[++p[0]]=i;
        for (int j=1;j<=p[0]&&p[j]*i<=maxn;j++)
            {pri[p[j]*i]=true;if (!(i%p[j])) break;}
    }
}
inline int Count(int n,int p) {int sum=0;while (n) sum+=n/p,n/=p;return sum;}
inline bignum Pow(bignum w,int b)
{
    static bignum s;s=1;
    while (b) {if (b&1) s=s*w;b>>=1;if (b) w=w*w;}
    return s;
}
int main()
{
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    scanf("%d",&n);if (n==1) return scanf("%d",&d[1]),putchar(!d[1]+48),0;
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&d[i]);if (!d[i]) return puts("0"),0;
        if (~d[i]) d[i]--,tot+=d[i]; else m++;
    }
    if (tot>n-2) return puts("0"),0;Make();ans=1;
    for (int i=1;i<=p[0];i++)
    {
        int ti=Count(n-2,p[i])-Count(n-2-tot,p[i]);
        for (int j=1;j<=n;j++) if (~d[j]) ti-=Count(d[j],p[i]);
        ans=ans*Pow(p[i],ti);
    }
    ans=ans*Pow(m,n-2-tot);
    return ans.print(),putchar('\n'),0;
}