[BZOJ3167][P4099][HEOI2013]SAO(树形DP)

题目描述

Welcome to SAO ( Strange and Abnormal Online)。这是一个 VR MMORPG， 含有 n 个关卡。但是，挑战不同关卡的顺序是一个很大的问题。

有 n – 1 个对于挑战关卡的限制，诸如第 i 个关卡必须在第 j 个关卡前挑战， 或者完成了第 k 个关卡才能挑战第 l 个关卡。并且，如果不考虑限制的方向性， 那么在这 n – 1 个限制的情况下，任何两个关卡都存在某种程度的关联性。即， 我们不能把所有关卡分成两个非空且不相交的子集，使得这两个子集之间没有任 何限制。

输入输出格式

输入格式：

第一行，一个整数 T，表示数据组数。

对于每组数据，第一行一个整数 n，表示关卡数。接下来 n – 1 行，每行为 “i sign j”，其中 0 ≤ i, j ≤ n – 1 且 i ≠ j，sign 为“<”或者“>”，表示第 i 个关卡 必须在第 j 个关卡前/后完成。

输出格式：

对于每个数据，输出一行一个整数，为攻克关卡的顺序方案个数，mod 1,000,000,007 输出。

输入输出样例

输入样例#1：
复制

2
5
0 < 2
1 < 2
2 < 3
2 < 4
4
0 < 1
0 < 2
0 < 3

输出样例#1： 复制

4
6

说明

对于 20%的数据有 n ≤ 10。

对于 40%的数据有 n ≤ 100。

对于另外 20%的数据有，保证数据中 sign 只会是<，并且 i < j。

对于 100%的数据有 T ≤ 5，1 ≤ n ≤ 1000。

洛谷上都有的题BZ怎么还权限啊。。

这题在某种意义上和 [HNOI2015]实验比较 有相似之处，主要思路是通过树形DP和组合数求解。

直接给出状转方程：$f[i][j]$表示以$i$为根的子树中，$i$排在第$j$位的方案数。乘上组合数即可。

https://www.cnblogs.com/CQzhangyu/p/7605703.html、

最后前缀和优化那一步可能不太一样。但都是$O(n^3)$的，不知道是怎么过的，应该是因为根本跑不满。

 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define mem(a) memset(a,0,sizeof(a))
 #define rep(i,l,r) for (int i=l; i<=r; i++)
 using namespace std;

 const int N=,mod=;
 char c;
 int n,T,u,v,cnt,C[N][N],g[N],f[N][N],to[N],sum[N][N],sz[N],h[N],nxt[N],val[N];
 void add(int u,int v,int w){ to[++cnt]=v; val[cnt]=w; nxt[cnt]=h[u]; h[u]=cnt; }
 void init(){ cnt=; mem(h); mem(f); mem(sum); mem(sz); }

 void dfs(int x,int fa){
     f[x][]=sum[x][]=sz[x]=;
     for (int i=h[x],s; i; i=nxt[i]) if ((s=to[i])!=fa){
         dfs(s,x); mem(g);
         if (val[i]==){
             rep(i,,sz[x]) if (f[x][i]) rep(j,,sz[s]) if (sum[s][j])
                 g[i+j]=(g[i+j]+1ll*f[x][i]*sum[s][j]%mod*C[i+j-][i-]%mod*C[sz[x]-i+sz[s]-j][sz[x]-i]%mod)%mod;
         }else{
             rep(i,,sz[x]) if (f[x][i]) rep(j,,sz[s])
                 g[i+j]=(g[i+j]+1ll*f[x][i]*(sum[s][sz[s]]-sum[s][j]+mod)%mod*C[i+j-][i-]%mod*C[sz[x]-i+sz[s]-j][sz[x]-i])%mod;
         }
         sz[x]+=sz[s];
         rep(i,,sz[x]) f[x][i]=g[i],sum[x][i]=(sum[x][i-]+g[i])%mod;
     }
 }

 int main(){
     freopen("bzoj3167.in","r",stdin);
     freopen("bzoj3167.out","w",stdout);
     rep(i,,) C[i][]=;
     rep(i,,) rep(j,,) C[i][j]=(C[i-][j-]+C[i-][j])%mod;
     for (scanf("%d",&T); T--; ){
         init(); scanf("%d",&n);
         rep(i,,n){
             scanf("%d %c%d",&u,&c,&v); u++; v++;
             if (c=='>') swap(u,v);
             add(u,v,); add(v,u,);
         }
         dfs(,); printf("%d\n",sum[][sz[]]);
     }
     return ;
 }