傳送門

---

值得注意的是一般的DAG的拓撲序列數量是NP問題，所以不能直接入手

題目中給出的圖可以看做是一個樹形圖，雖然方向比較迷。考慮使用樹形圖的性質

不妨任選一個點為根做樹形DP，注意到數的位置與方案數相關，所以也要設在狀態內。故設\(f_{i,j}\)表示對于\(i\)及\(i\)的子樹所有點構成的拓撲序列，\(i\)排在第\(j\)位的方案數，通過一個兒子一個兒子地合并來轉移。

對于當前計算的點\(u\)的某一個兒子\(v\)已經算完，正要和\(u\)合并。設\(sz_u\)表示\(u\)和\(u\)的已合并子樹的點數之和，并設\(u < v\)，那么序列中\(v\)要在\(u\)的后面。

先枚舉\(u\)在已合并序列中的位置\(j\)，然后枚舉\(v\)的子樹對應的拓撲序列中有\(k\)個放在\(u\)的前面，不難得到轉移方程：\(dp_{u,j+k} \leftarrow dp_{u,j} \times C_{j-1+k}^k \times C_{sz_u + sz_v - j - k} ^ {sz_v - k} \times \sum\limits_{i=k+1}^{sz_v} dp_{v,k}\)

如果\(u>v\)也是差不多的

前后綴和優化最后的\(\sum dp_{v,k}\)并預處理組合數

關于復雜度，注意到每一次合并的復雜度為"已合并的兒子大小"\(\times\)"當前合并的兒子大小"，可以看作任意兩個點只會在它們的LCA處產生\(1\)的復雜度，所以總復雜度是\(O(n^2)\)的。

#include<iostream> #include<cstdio> #include<cstdlib> #include<ctime> #include<cctype> #include<algorithm> #include<cstring> #include<iomanip> #include<queue> #include<map> #include<set> #include<bitset> #include<stack> #include<vector> #include<cmath> #include<cassert> //This code is written by Itst using namespace std;inline int read(){int a = 0;char c = getchar();bool f = 0;while(!isdigit(c) && c != EOF){if(c == '-')f = 1;c = getchar();}if(c == EOF)exit(0);while(isdigit(c)){a = a * 10 + c - 48;c = getchar();}return f ? -a : a; }const int MOD = 1e9 + 7; int N , cntEd; struct Edge{int end , upEd; }Ed[2003]; int head[1007] , C[1007][1007] , dp[1007][1007] , sz[1007] , pre[1007][1007] , suf[1007][1007] , tmp[1007];inline void addEd(int a , int b){Ed[++cntEd].end = b;Ed[cntEd].upEd = head[a];head[a] = cntEd; }inline char getc(){char c = getchar();while(c == ' ' || c == '\n' || c == '\r')c = getchar();return c; }void init(){C[0][0] = 1;for(int i = 1 ; i <= 1000 ; ++i){C[i][0] = 1;for(int j = 1 ; j <= 1000 ; ++j)C[i][j] = (C[i - 1][j] + C[i - 1][j - 1]) % MOD;} }void dfs(int u , int p){sz[u] = 1;dp[u][1] = 1;for(int i = head[u] ; i ; i = Ed[i].upEd)if(Ed[i].end != p){dfs(Ed[i].end , u);int v = Ed[i].end;sz[u] += sz[v];memset(tmp , 0 , sizeof(tmp));if(i & 1)for(int j = sz[u] - sz[v] ; j ; --j)for(int k = 0 ; k < sz[v] ; ++k)tmp[j + k] = (tmp[j + k] + 1ll * dp[u][j] * suf[v][k + 1] % MOD * C[j + k - 1][k] % MOD * C[sz[u] - j - k][sz[v] - k]) % MOD;elsefor(int j = sz[u] - sz[v] ; j ; --j)for(int k = 1 ; k <= sz[v] ; ++k)tmp[j + k] = (tmp[j + k] + 1ll * dp[u][j] * pre[v][k] % MOD * C[j + k - 1][k] % MOD * C[sz[u] - j - k][sz[v] - k]) % MOD;memcpy(dp[u] , tmp , sizeof(tmp));}for(int i = 1 ; i <= sz[u] ; ++i)pre[u][i] = (pre[u][i - 1] + dp[u][i]) % MOD;for(int i = sz[u] ; i ; --i)suf[u][i] = (suf[u][i + 1] + dp[u][i]) % MOD; }int main(){ #ifndef ONLINE_JUDGEfreopen("in","r",stdin);//freopen("out","w",stdout); #endifinit();for(int T = read() ; T ; --T){N = read();for(int i = 1 ; i < N ; ++i){int a = read() + 1;char c = getc();int b = read() + 1;if(c == '>')swap(a , b);addEd(a , b);addEd(b , a);}dfs(1 , 0);cout << pre[1][N] << endl;memset(dp , 0 , sizeof(dp));memset(pre , 0 , sizeof(pre));memset(suf , 0 , sizeof(suf));memset(head , 0 , sizeof(head));cntEd = 0;}return 0; } =

轉載于:https://www.cnblogs.com/Itst/p/10406642.html

總結

以上是生活随笔為你收集整理的Luogu4099 HEOI2013 SAO 组合、树形DP的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。