【BZOJ3167】[HEOI2013]SAO（动态规划）
题面
题解
显然限制条件是一个\(DAG\)（不考虑边的方向的话就是一棵树了）。
那么考虑树型\(dp\)，设\(f[i][j]\)表示当前点\(i\)在其子树内的排名为\(j\)的方案数。
每次考虑加入一棵子树，即考虑把\(f[v][k]\)加入到\(f[u][i]\)的贡献中。
分类讨论，如果\(v\)应当在\(u\)之前，枚举\(v\)的子树内一共有多少个点在\(u\)之前，那么假设\(u\)当前的序列长度为\(x\)，\(v\)当前的序列长度为\(y\)，枚举一共有\(j\)个\(v\)的子树内的点在\(u\)之前。
那么此时的方案数就是:\(f[u][i+j]\leftarrow {i+j-1\choose i-1}*f[u][i]*f[v][k]*{y-j+x-i\choose x-i}\)
然后我们来学习枚举（来写假的pascal代码）
for i:=1 to x do
for k:=1 to y do
for j:=k to y do
f[u][i+j]+=f[u][i]*f[v][k]*C[i+j-1][i-1]*C[x+y-i-j][x-i]
似乎这个\(k\)只在\(f[v][k]\)中出现过？？？
那么我们换一下
for i:=1 to x do
for j:=1 to y do
for k:=1 to j do
f[u][i+j]+=f[u][i]*f[v][k]*C[i+j-1][i-1]*C[x+y-i-j][x-i]
发现把\(f[v]\)数组前缀和一下底下这个玩意就可以做到\(O(1)\)转移啦。
而如果\(v\)要放在\(u\)之后也是类似的。
还是枚举有多少个放在\(u\)之前，那么这一部分的贡献是没变的，只是转移的时候的范围要变一变。
for i:=1 to x do
for k:=1 to y do
for j:=0 to k-1 do
f[u][i+j]+=f[u][i]*f[v][k]*C[i+j-1][i-1]*C[x+y-i-j][x-i]
还是换一下
for i:=1 to x do
for j:=0 to y do
for k:=i+1 to y do
f[u][i+j]+=f[u][i]*f[v][k]*C[i+j-1][i-1]*C[x+y-i-j][x-i]
这样子上面两层循环枚举的都是子树大小，因此全局总的复杂度就是\(O(n^2)\)啦。
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
#define MOD 1000000007
#define MAX 1010
void add(int &x,int y){x+=y;if(x>=MOD)x-=MOD;}
inline int read()
{
int x=0;bool t=false;char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=true,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return t?-x:x;
}
struct Line{int v,next,w;}e[MAX<<1];
int h[MAX],cnt;
inline void Add(int u,int v,int w){e[cnt]=(Line){v,h[u],w};h[u]=cnt++;}
void init(){memset(h,0,sizeof(h));cnt=1;}
int n,C[MAX][MAX];char ch[2];
int f[MAX][MAX],sz[MAX],tmp[MAX];
void dfs(int u,int ff)
{
f[u][1]=1;sz[u]=1;
for(int i=h[u];i;i=e[i].next)
{
int v=e[i].v;if(v==ff)continue;
dfs(e[i].v,u);
for(int a=1;a<=sz[u];++a)
for(int b=0;b<=sz[v];++b)
if(e[i].w)
add(tmp[a+b],1ll*f[u][a]*f[v][b]%MOD*C[a+b-1][a-1]%MOD*C[sz[u]+sz[v]-a-b][sz[u]-a]%MOD);
else
add(tmp[a+b],1ll*f[u][a]*(f[v][sz[v]]-f[v][b]+MOD)%MOD*C[a+b-1][a-1]%MOD*C[sz[u]+sz[v]-a-b][sz[u]-a]%MOD);
sz[u]+=sz[v];
for(int j=1;j<=sz[u];++j)f[u][j]=tmp[j],tmp[j]=0;
}
for(int j=1;j<=sz[u];++j)add(f[u][j],f[u][j-1]);
}
int main()
{
for(int i=0;i<MAX;++i)C[i][0]=1;
for(int i=1;i<MAX;++i)
for(int j=1;j<=i;++j)C[i][j]=(C[i-1][j]+C[i-1][j-1])%MOD;
int T=read();
while(T--)
{
init();n=read();
for(int i=1,u,v;i<n;++i)
{
u=read()+1;scanf("%s",ch);v=read()+1;
if(ch[0]=='<')Add(u,v,0),Add(v,u,1);
else Add(u,v,1),Add(v,u,0);
}
dfs(1,0);
printf("%d\n",f[1][n]);
}
return 0;
}