隐形斗篷

隐形斗篷

题解

因为我们现在要求的不同的树的定义是点集不同或连边方式不同，也就是无根树的形态不同，所以我们很快就想到了构造prufer序列。
由于prufer序列与树的形态是意义对应的，所以我们要求的就是合法的prufer序列数量。
对于点$i$的度数不超过$a_i$的条件我们可以用在prufer序列中的出现次数不超过$a_i-1$次来维护。
当然，如果我们直接暴力枚举prufer序列是明显不行的。
设我们在prufer序列中出现了$k$个点，每个点的出现次数为$b_i$，树的大小为$x$，那么这样的prufer序列个数为$\frac{x!}{{\prod }_{i=1}^k{b}_i!}$，其能够构成的树的个数为$\left(\genfrac{}{}{0ex}{}{n-k}{x-k}\right)\frac{x!}{{\prod }_{i=1}^k{b}_i!}$，需要乘上除这些点以外其它点的选择情况。
由于prufer序列本身的构成状况是从小到大选的，所以没必要去考虑选出来的点的顺序。
我们很快的发现如果这样写的话，我们是可以通过dp求出prufer序列的个数的。
记$f_{i,j,k}$表示前$i$个点中prufer序列的点集中出现了$j$个，且这些点在序列中的总出现次数为$k$的序列个数，顺便将表示这些点相对顺序的$\frac{1}{b_i!}$也可以加进去。
这样，我们就很快得到了状态转移方程式，
$$f_{i,j,k}=f_{i-1,j,k}+\sum _{x=1}^{a_i-1}\frac{f_{i-1,j-1,k-x}}{x!}$$
这不就是指数生成函数吗？
将最后一位变成多项式$x^k$次方的系数，有
$f_{i,j}=f_{i-1,j}+f_{i-1,j-1}\times ({\sum }_{k=1}^{a_i-1}\frac{x^k}{k!})$
我们需要的只是前面$n$个，$x^n$后的可以省去。
最后答案为$(x-2)!{\sum }_{i=1}^{x-2}\left(\genfrac{}{}{0ex}{}{n-i}{x-i}\right)f_{n,i,x-2}$
将$f$全部求出来就可以了。

这样，我们就成功将时间复杂度优化到了$O\left(T{n}^{3}logn\right)$
但我们很快就发现一个惊人的问题$mod=1e9+7$！
莫非我们要打多模数NTT？其实可以暴力转移的。
时间复杂度$O\left(n^4\right)$也不会T。

源码

#include<cstdio>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<time.h>
using namespace std;
#define MAXN 100005
#define lowbit(x) (x&-x)
#define reg register
typedef long long LL;
typedef unsigned long long uLL;
typedef unsigned int uint;
typedef pair<int,int> pii;
const int INF=0x7f7f7f7f;
const int mo=1e9+7;
const double PI=acos(-1.0);
template<typename _T>
_T Fabs(_T x){return x<0?-x:x;}
template<typename _T>
void read(_T &x){
	_T f=1;x=0;char s=getchar();
	while(s>'9'||s<'0'){if(s=='-')f=-1;s=getchar();}
	while('0'<=s&&s<='9'){x=(x<<3)+(x<<1)+(s^48);s=getchar();}
	x*=f;
}
int add(int x,int y){return x+y<mo?x+y:x+y-mo;}
int qkpow(int a,int s){int t=1;while(s){if(s&1)t=1ll*a*t%mo;a=1ll*a*a%mo;s>>=1;}return t;}
int T,n;
struct ming{
	int a[55];ming(){memset(a,0,sizeof(a));}void clear(){memset(a,0,sizeof(a));}
	ming friend operator + (ming A,ming B){ming C;for(int i=0;i<=n;i++)C.a[i]=add(A.a[i],B.a[i]);return C;}
	ming friend operator * (ming A,ming B){ming C;for(int i=0;i<=n;i++)for(int j=0;j<=i;j++)C.a[i]=add(C.a[i],1ll*A.a[j]*B.a[i-j]%mo);return C;}
}dp[55][55],F[55];
int a[55],fac[55],inv[55],f[55];
int C(int x,int y){if(x<0||y<0||x<y)return 0;return 1ll*fac[x]*inv[y]%mo*inv[x-y]%mo;}
void init(){
	fac[0]=fac[1]=inv[0]=inv[1]=f[1]=F[1].a[1]=1;
	for(int i=2;i<=50;i++){
		fac[i]=1ll*i*fac[i-1]%mo;f[i]=1ll*(mo-mo/i)*f[mo%i]%mo;
		inv[i]=1ll*f[i]*inv[i-1]%mo;F[i]=F[i-1];F[i].a[i]=inv[i];
	}
}
signed main(){
	freopen("c.in","r",stdin);
	freopen("c.out","w",stdout);
	read(T);init();
	while(T--){
		read(n);for(int i=1;i<=n;i++)read(a[i]);
		for(int i=0;i<=n;i++)for(int j=0;j<=i;j++)dp[i][j].clear();dp[0][0].a[0]=1;
		for(int i=1;i<=n;i++)for(int j=0;j<=i;j++)dp[i][j]=dp[i-1][j-1]*F[a[i]-1]+dp[i-1][j];
		printf("%d %d ",n,C(n,2));
		for(int i=3;i<=n;i++){
			int tmp=0;
			for(int j=1;j<i-1;j++)
				tmp=add(1ll*C(n-j,i-j)*dp[n][j].a[i-2]%mo,tmp);
			printf("%d ",1ll*fac[i-2]*tmp%mo);
		}
		puts("");
	}
	return 0;
}

谢谢！！！