3167: [Heoi2013]Sao [树形DP]

3167: [Heoi2013]Sao

题意：

n个点的“有向”树，求拓扑排序方案数

---

Welcome to Sword Art Online!!!

一开始想错了...没有考虑一个点的孩子可以排在父亲后...

为了能转移，给状态加一维，\(f[i][j]\)表示子树i，i排在第j位的方案数

然后，很像树形背包啊，转移枚举孩子子树中k个点在i之前，更新\(f[i][j+k]\)

严格做到每次合并复杂度为 “已经合并大小*正要合并进去的大小”，那么这个复杂度就是\(O(n^2)\)的，因为一个点对只贡献一次

真不敢相信我以前的树形背包都写的是\(O(n^3)\)

注意需要用一个新数组保存当前更新得到的值

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
using namespace std;
typedef long long ll;
const int N = 1005, P = 1e9+7;
inline int read() {
    char c=getchar(); int x=0,f=1;
    while(c<'0'||c>'9') {if(c=='-')f=-1;c=getchar();}
    while(c>='0'&&c<='9') {x=x*10+c-'0';c=getchar();}
    return x*f;
}

int n, u, v, c[N][N];
char s[5];
struct edge{int v, ne, p;} e[N<<1];
int cnt, h[N];
inline void ins(int u, int v) { //printf("ins %d %d\n", u, v);
	e[++cnt] = (edge){v, h[u], 1}; h[u] = cnt;
	e[++cnt] = (edge){u, h[v], 0}; h[v] = cnt;
}
int f[N][N], size[N], f_pre[N][N], f_suf[N][N], g[N];
void dp(int u, int fa) {
	size[u] = 1; f[u][1] = 1;
	for(int i=h[u]; i; i=e[i].ne) {
		int v = e[i].v, p = e[i].p;
		if(v == fa) continue;
		dp(v, u);
		for(int j = 0; j <= size[u] + size[v]; j++) g[j] = 0;
		for(int j = 1; j <= size[u]; j++)
			for(int k = 0; k <= size[v]; k++) {
				ll t = p ? f_suf[v][k+1] : f_pre[v][k];
				g[j+k] = (g[j+k] + (ll) c[j-1+k][k] * c[size[u]-j+size[v]-k][size[v]-k] %P * f[u][j] %P * t) %P;
			}
		for(int j = 0; j <= size[u] + size[v]; j++) f[u][j] = g[j];
		size[u] += size[v];
	}
	for(int i = 1; i <= size[u]; i++) f_pre[u][i] = (f_pre[u][i-1] + f[u][i]) %P;
	for(int i = size[u]; i >= 1; i--) f_suf[u][i] = (f_suf[u][i+1] + f[u][i]) %P;
}
int main() {
    freopen("in", "r", stdin);
	int T = read();
	c[0][0] = 1;
	for(int i=1; i<N; i++) {
		c[i][0] = 1;
		for(int j=1; j<N; j++) c[i][j] = (c[i-1][j] + c[i-1][j-1]) %P;
	}
	while(T--) {
		n = read();
		memset(f, 0, sizeof(f));
		memset(f_pre, 0, sizeof(f_pre));
		memset(f_suf, 0, sizeof(f_suf));
		cnt = 0; memset(h, 0, sizeof(h));
		for(int i=1; i<n; i++) {
			u = read()+1; scanf("%s", s); v = read()+1;
			if(s[0] == '<') ins(u, v); else ins(v, u);
		}
		dp(1, 0);
		printf("%d\n", f_pre[1][size[1]]);
	}
}