bzoj 1211[HNOI2004]树的计数

题目链接

prufer序列可以知道度数为d的点会出现d-1次,而一共有n-2 个数出现,

显然可以由排列公式得知:ans=(n−2)!∏ni=1(di−1)!

Ps:乘法中会爆掉long long可以用唯一分解定理表示我会说这样预处理,我跑了bzoj倒数第二!!!,也可以用多个数的乘积表示.注意对不成立情况的讨论

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<vector>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fod(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
typedef long long ll;
const ll N=1e6+10;
vector<int>pri;vector<int>tm;
bool check
;ll d[200],n,tot=0,flag=1;
void getprime()
{
memset(check,0,sizeof(check));
for(int i=2;i<=N;i++) {
if(!check[i]) pri.push_back(i),tm.push_back(0);
for(int j=0;j<int(pri.size());j++) {
if(pri[j]*i>N) break;
check[i*pri[j]]=1;
if(i%pri[j]==0) break;
}
}
}
void Dec(ll x,int data)
{
if(x==0||x==1) return ;
for(int i=0;i<int(pri.size());i++) {
if(x==1) break;
while(x%pri[i]==0) {
x/=pri[i];
tm[i]+=data;
}
}
return ;
}
ll qpow(ll a,ll b)
{
ll ret=1;
for(ll i=b;i;i>>=1,a*=a)
if(i&1) ret*=a;
return ret;
}
ll solve()
{
if(tot!=n-2||!flag||(n==1&&d[1]==1)) return 0;
for(int i=1;i<=n-2;i++)
Dec(i,1);
for(int i=1;i<=n;i++) {
for(int j=d[i]-1;j>=0;j--) {
Dec(j,-1);
}
}
ll ans=1;
for(int i=0;i<int(pri.size());i++)
ans*=qpow(pri[i],tm[i]);
return ans;
}
int main()
{
getprime();
scanf("%lld",&n);
fo(i,1,n) {scanf("%lld",&d[i]);tot+=d[i]-1;if(!d[i]&&n!=1) flag=0;}
printf("%lld\n",solve());
return 0;
}