BZOJ 1005 [HNOI2008]明明的烦恼

最新推荐文章于 2018-08-21 15:27:00 发布

wang3312362136

最新推荐文章于 2018-08-21 15:27:00 发布

阅读量95

点赞数

分类专栏：计数

本文链接：https://blog.csdn.net/wang3312362136/article/details/80540119

版权

计数专栏收录该内容

23 篇文章 0 订阅

订阅专栏

题目链接

https://www.lydsy.com/JudgeOnline/problem.php?id=1005

思路

由于这是无根树，因此我们把树转化成prufer序列，由于prufer序列的性质，很容易得到满足条件prufer序列的个数为：

(n - 2 s u m) s u m ! \prod c n t i = 1 ( d i - 1 ) ! \cdot (n - c n t) n - 2 - s u m

$\binom{n-2}{sum}\frac{sum!}{\prod_{i=1}^{cnt}(d_i-1)!}\cdot (n-cnt)^{n-2-sum}$
此处

sum=∑ni=1[di≠−1]×(di−1),cnt=∑ni=1[di≠−1] s u m = ∑ i = 1 n [ d i ≠ − 1 ] × ( d i − 1 ) , c n t = ∑ i = 1 n [ d i ≠ − 1 ] $sum=\sum_{i=1}^n [d_i\not= -1]\times (d_i-1),cnt=\sum_{i=1}^n [d_i\not= -1]$

化简得到：

( n - 2 ) ! ( n - 2 - s u m ) ! \prod c n t i = 1 ( d i - 1 ) ! \cdot (n - c n t) n - 2 - s u m

$\frac{(n-2)!}{(n-2-sum)!\prod_{i=1}^{cnt}(d_i-1)!}\cdot (n-cnt)^{n-2-sum}$
注意这道题要高精度乘法。

代码

#include <cstdio>
#include <cmath>
#include <algorithm>

const int maxn=1000;
const int maxd=100000;

int read()
{
  int x=0,f=1;
  char ch=getchar();
  while((ch<'0')||(ch>'9'))
    {
      if(ch=='-')
        {
          f=-f;
        }
      ch=getchar();
    }
  while((ch>='0')&&(ch<='9'))
    {
      x=x*10+ch-'0';
      ch=getchar();
    }
  return x*f;
}

int n,d[maxn+10],p[maxn+10][maxn+10],ans[maxn+10];
int bignum[maxd+10],hi,tot,cnt,sum;

inline int times(int v)
{
  int r=0;
  for(int i=1; i<=hi; ++i)
    {
      bignum[i]=bignum[i]*v+r;
      r=bignum[i]/10;
      bignum[i]%=10;
    }
  while(r)
    {
      bignum[++hi]=r%10;
      r/=10;
    }
  return 0;
}

int main()
{
  n=read();
  if(n==1)
    {
      int x=read();
      if(x<=0)
        {
          puts("1");
        }
      else
        {
          puts("0");
        }
      return 0;
    }
  for(int i=1; i<=n; ++i)
    {
      d[i]=read();
      if(!d[i])
        {
          puts("0");
          return 0;
        }
      else if(d[i]==-1)
        {
          tot=-1;
        }
      else
        {
          if(tot!=-1)
            {
              tot+=d[i];
            }
          sum+=d[i]-1;
          ++cnt;
        }
    }
  if((tot!=-1)&&(tot!=2*n-2))
    {
      puts("0");
      return 0;
    }
  for(int i=2; i<=n; ++i)
    {
      int v=i;
      for(int j=2; j<=sqrt(i); ++j)
        {
          while(v%j==0)
            {
              v/=j;
              ++p[i][j];
            }
          if(v==1)
            {
              break;
            }
        }
      if(v!=1)
        {
          ++p[i][v];
        }
    }
  for(int i=2; i<=n; ++i)
    {
      for(int j=2; j<=i; ++j)
        {
          p[i][j]+=p[i-1][j];
        }
    }
  for(int i=2; i<=n-2; ++i)
    {
      ans[i]=p[n-2][i]-p[n-2-sum][i];
    }
  for(int i=1; i<=n; ++i)
    {
      if(d[i]==-1)
        {
          continue;
        }
      for(int j=2; j<=d[i]-1; ++j)
        {
          ans[j]-=p[d[i]-1][j];
        }
    }
  hi=1;
  bignum[1]=1;
  for(int i=2; i<=n; ++i)
    {
      for(int j=1; j<=ans[i]; ++j)
        {
          times(i);
        }
    }
  for(int i=1; i<=n-2-sum; ++i)
    {
      times(n-cnt);
    }
  for(int i=hi; i; --i)
    {
      printf("%d",bignum[i]);
    }
  puts("");
  return 0;
}