HEOI 2013 SAO 题解

题目传送门

题目大意： 给出一张 $n-1$ 条边 $n$ 个点的弱连通图，求拓扑排序数。

题解

设 $f_{i,j}$ 表示在 $i$ 的子树的拓扑序中，$i$ 在第 $j$ 位的方案数。

考虑一个节点 $x$ 要如何与儿子 $y$ 的信息进行合并。假设 $x$ 的拓扑序要在 $y$ 前面，枚举 $i,j,k$ 表示 $x$ 原本在拓扑序第 $i$ 位，$y$ 原本在拓扑序第 $j$ 位，两个序列合并后，$x$ 在拓扑序第 $k$ 位。

由于原本拓扑序在 $x$ 前面的节点，合并后不可能跑到 $x$ 后面，所以这 $i-1$ 个节点肯定合并后在前 $k-1$ 中，用一个组合数选出方案数 $\left(\genfrac{}{}{0ex}{}{k-1}{i-1}\right)$。

同样的，$x$ 后面的节点依然在 $x$ 后面，有 $\left(\genfrac{}{}{0ex}{}{s{z}_x+s{z}_y-k}{s{z}_x-i}\right)$。

再考虑一下 $k$ 的取值范围，首先 $k$ 肯定不小于 $i$，若 $y$ 子树内拓扑序比 $y$ 小的那些全部都比 $x$ 小，那么此时 $k$ 取到最大值 $i+j-1$。

于是得到转移方程：$f_{x,k}^{\prime }={\sum }_{k=i}^{i+j-1}\left(\genfrac{}{}{0ex}{}{k-1}{i-1}\right)\left(\genfrac{}{}{0ex}{}{s{z}_x+s{z}_y-k}{s{z}_x-i}\right)f_{x,i}{f}_{y,j}$。

这样是 $O(n^3)$ 的，考虑优化。发现 $j$ 在方程中只出现过一次，假如最后枚举 $j$，就可以用前缀和优化了。

具体来说，原来的枚举顺序是：

for(int i=1;i<=sz[x];i++)
	for(int j=1;j<=sz[y];j++)
		for(int k=i;k<=i+j-1;k++)	

变成：

for(int i=1;i<=sz[x];i++)
	for(int k=i;k<=i+sz[y]-1;k++)
		for(int j=k-i+1;j<=sz[y];j++)

$x$ 的拓扑序在 $y$ 后面的情况类似，将 $k$ 的枚举范围改成 $i+j$ ~ $i+s{z}_y$ 即可，前缀和优化部分的枚举变成：

for(int i=1;i<=sz[x];i++)
	for(int k=i+1;k<=i+sz[y];k++)
		for(int j=1;j<=k-i;j++)

代码如下：

#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
#define maxn 1010
#define mod 1000000007
#define pb push_back

int T,n;
vector<int> e[maxn],w[maxn];
int fac[maxn],inv_fac[maxn],inv[maxn];
void FacInit(){
	fac[0]=inv_fac[0]=inv[1]=1;
	for(int i=1;i<=maxn-10;i++)fac[i]=1ll*fac[i-1]*i%mod;
	for(int i=2;i<=maxn-10;i++)inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
	for(int i=1;i<=maxn-10;i++)inv_fac[i]=1ll*inv_fac[i-1]*inv[i]%mod;
}
int Binom(int x,int y){return 1ll*fac[x]*inv_fac[y]%mod*inv_fac[x-y]%mod;}
int sz[maxn],f[maxn][maxn],tmp[maxn];
void add(int &x,int y){x=(x+y>=mod?x+y-mod:x+y);}
void dfs(int x,int fa){
	f[x][1]=1;sz[x]=1;
	for(int d=0,y;d<e[x].size();d++)if((y=e[x][d])!=fa){
		dfs(y,x);
		for(int i=1;i<=sz[x];i++){
			if(!w[x][d]){
				for(int k=i;k<=i+sz[y]-1;k++)
				add(tmp[k],1ll*Binom(k-1,i-1)*Binom(sz[x]+sz[y]-k,sz[x]-i)%mod*f[x][i]%mod*(f[y][sz[y]]-f[y][k-i]+mod)%mod);
			}else{
				for(int k=i+1;k<=i+sz[y];k++)
				add(tmp[k],1ll*Binom(k-1,i-1)*Binom(sz[x]+sz[y]-k,sz[x]-i)%mod*f[x][i]%mod*(f[y][k-i]-f[y][0]+mod)%mod);
			}
		}
		sz[x]+=sz[y];
		for(int i=1;i<=sz[x];i++)f[x][i]=tmp[i],tmp[i]=0;
	}
	for(int i=1;i<=sz[x];i++)add(f[x][i],f[x][i-1]);
}

int main()
{
	FacInit();
	scanf("%d",&T);while(T--)
	{
		scanf("%d",&n);
		for(int i=1;i<=n;i++)e[i].clear(),w[i].clear();
		for(int i=1;i<n;i++){
			int x,y;char s[2];
			scanf("%d %s %d",&x,s,&y);
			int p=s[0]!='<';x++;y++;
			e[x].pb(y);w[x].pb(p);
			e[y].pb(x);w[y].pb(p^1);
		}
		dfs(1,0);
		printf("%d\n",f[1][n]);
	}
}