码迷,mamicode.com
首页 > 其他好文 > 详细

【bzoj1211】 HNOI2004—树的计数

时间:2016-09-27 22:52:47      阅读:213      评论:0      收藏:0      [点我收藏+]

标签:

http://www.lydsy.com/JudgeOnline/problem.php?id=1211 (题目链接)

题意:一个有n个结点的树,设它的结点分别为v1, v2, …, vn,已知第i个结点vi的度数为di,问满足这样的条件的不同的树有多少棵。给定n,d1, d2, …, dn,编程需要输出满足d(vi)=di的树的个数。

Solution 
  prufer序列,明明的烦恼简化版。

代码:

// bzoj1211
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<cstdio>
#include<cmath>
#include<map>
#define inf 2147483640
#define LL long long
#define free(a) freopen(a".in","r",stdin);freopen(a".out","w",stdout);
using namespace std;
inline LL getint() {
    LL x=0,f=1;char ch=getchar();
    while (ch>‘9‘ || ch<‘0‘) {if (ch==‘-‘) f=-1;ch=getchar();}
    while (ch>=‘0‘ && ch<=‘9‘) {x=x*10+ch-‘0‘;ch=getchar();}
    return x*f;
}

const int maxn=200;
int d[maxn],a[maxn],np[maxn],n;

void pls(int x,int f) {
    for (int i=2;i<=x;i++)
        if (!np[i]) for (int j=i;j<=x;j*=i) a[i]+=f*x/j;
}
int main() {
    np[1]=1;
    for (int i=2;i<=150;i++)
        if (!np[i]) for (int j=i*2;j<=150;j+=i) np[j]=1;
    scanf("%d",&n);
    int sum=0;
    if (n==1) {
        int x;scanf("%d",&x);
        if (!x) printf("1");
        else printf("0");
        return 0;
    }
    for (int i=1;i<=n;i++) {
        scanf("%d",&d[i]);
        if (!d[i]) {printf("0");return 0;}
        d[i]--;
        sum+=d[i];
        pls(d[i],-1);
    }
    if (sum!=n-2) {printf("0");return 0;}
    LL ans=1;
    pls(sum,1);
    for (int i=2;i<=150;i++)
        for (int j=1;j<=a[i];j++) ans*=(LL)i;
    printf("%lld",ans);
    return 0;
}

  

【bzoj1211】 HNOI2004—树的计数

标签:

原文地址:http://www.cnblogs.com/MashiroSky/p/5914432.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!