HEOI2013SAO

题目描述

给定一个(DAG),问这个(DAG)有多少种拓扑序。

题解

我们首先需要设计一个能够比较好的转移的状态。

我们可以设(dp[i][j])表示第i个点在当前(dp)的子图中拓扑排名为(j)的方案数。

至于(dp)的方式,我们发现只有(n-1)条边,所以我们并不用在(DAG)(dp),直接在建出来的树上(dp)就好了。

转移的话,对于一条树边(u->v)我们先枚举转移之后的(u)的排名,再枚举当前(v)的排名,我们发现还需要枚举(v)的子树在(u)排名之前的点的个数(x)

(u)需要在(v​)之后时:

[dp[u][j]=sum_{k=1}^{kleq size[v]}sum_{x=k}^{xleq size[v]}dp[u][j-x]*inom{j-1}{x}*inom{size[u]-(j-x)+size[v]-x}{size[v]-x} ]

发现转移是一个后缀和的形式,可以用后缀和优化)。

(u)需要在(v)之前时。

[dp[u][j]=sum_{k=1}^{kleq size[v]}sum_{x=1}^{x<k}dp[u][j-x]*inom{j-1}{x}*inom{size[u]-(j-x)+size[v]-x}{size[v]-x} ]

转移是一个前缀和的形式,也可以用前缀和优化。

经过优化后,这个(dp)的复杂度可以优化到树形背包的复杂度,也就是(O(n^2))

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define N 1009
using namespace std;
typedef long long ll;
const int maxn=1000;
const int mod=1000000007;
char s[1];
ll ans,dp[N][N],now[N],c[N][N],g[N];
int tot,head[N],size[N],n;
inline ll rd(){
	ll x=0;char c=getchar();bool f=0;
	while(!isdigit(c)){if(c=='-')f=1;c=getchar();}
	while(isdigit(c)){x=(x<<1)+(x<<3)+(c^48);c=getchar();}
	return f?-x:x;
}
inline void MOD(ll &x){x=x>=mod?x-mod:x;}
struct edge{int n,to,l;}e[N<<1];
inline void add(int u,int v,int l){e[++tot].n=head[u];e[tot].to=v;head[u]=tot;e[tot].l=l;}
void dfs(int u,int fa){
	size[u]=1; 
	dp[u][1]=1;
	for(int i=head[u];i;i=e[i].n){
		int v=e[i].to;if(v==fa)continue;
		dfs(v,u);
		if(e[i].l){
		    for(int j=1;j<=size[u]+size[v];++j){
		      int ma=min(size[v],j-1);		      
			  now[ma+1]=0;
		      for(int x=ma;x>=1;--x)
		      	MOD(now[x]=now[x+1]+dp[u][j-x]*c[j-1][x]%mod*c[size[u]-(j-x)+size[v]-x][size[v]-x]%mod);
			  for(int k=1;k<=size[v];++k)MOD(g[j]+=dp[v][k]*now[k]%mod),now[k]=0;
		    }
		}
		else{
			for(int j=1;j<=size[u]+size[v];++j){
			   int ma=min(size[v],j-1);
			   for(int x=0;x<=ma;++x)
			     MOD(now[x]=now[x-1]+dp[u][j-x]*c[j-1][x]%mod*c[size[u]-(j-x)+size[v]-x][size[v]-x]%mod);
			   for(int k=1;k<=size[v];++k)MOD(g[j]+=now[min(k-1,ma)]*dp[v][k]%mod);
			   for(int x=0;x<=ma;++x)now[x]=0;
		    }
		}
		size[u]+=size[v];
		for(int j=1;j<=size[u];++j)dp[u][j]=g[j],g[j]=0;
	}
}
int main(){   // 1 -> dayu 0 -> xiaoyu
	int T=rd();
    c[0][0]=1;
    for(int i=1;i<=maxn;++i){
    	c[i][0]=1;
    	for(int j=1;j<=i;++j)MOD(c[i][j]=c[i-1][j]+c[i-1][j-1]);
	}
	while(T--){
		n=rd();int u,v;
		memset(dp,0,sizeof(dp));
		memset(head,0,sizeof(head));tot=0;
		for(int i=1;i<n;++i){
			u=rd()+1;scanf("%s",s);v=rd()+1;
			add(u,v,s[0]=='>');add(v,u,s[0]=='<');
		}
		ans=0;dfs(1,0);
		for(int i=1;i<=n;++i)MOD(ans+=dp[1][i]);
	    printf("%lld
",ans);
	}
	return 0;
}
原文地址:https://www.cnblogs.com/ZH-comld/p/10542728.html