洛谷1040 加分二叉樹
阿新 • • 發佈:2018-12-26
原題連結
挺水的一道區間\(DP\)。
設\(f[i][j]\)表示在中序遍歷下編號\(i \sim j\)的點所構成的子樹的最高加分,列舉\(k\)為子樹的根,則有狀態轉移方程:
\[f[i][j] = \max \limits _{k = i + 1} ^ {j - 1} \{ f[i][k - 1] \times f[k + 1][j] + f[k][k] \}\]
初始化\(f[i][i]\)為點\(i\)的分數,\(f[i][j] = f[i][i] + f[i + 1][j]\),即左子樹為空。
因為題目並沒有說清楚,實際上要求的前序遍歷是所有解中字典序最小的,所以我們要儘量使得編號小的作為根或是左子樹中的點,按編號從小到大列舉哪個作為根即可。
求前序遍歷可以定義\(r[i][j]\)
#include<cstdio> using namespace std; typedef long long ll; const int N = 50; ll f[N][N]; int r[N][N]; inline int re() { int x = 0; char c = getchar(); bool p = 0; for (; c < '0' || c > '9'; c = getchar()) p |= c == '-'; for (; c >= '0' && c <= '9'; c = getchar()) x = x * 10 + c - '0'; return p ? -x : x; } void pr(int x, int y) { if (x > y) return; printf("%d ", r[x][y]); pr(x, r[x][y] - 1); pr(r[x][y] + 1, y); } int main() { int i, j, k, l, n; n = re(); for (i = 1; i <= n; i++) f[i][i] = re(), r[i][i] = i; for (l = 2; l <= n; l++) for (i = 1; i + l - 1 <= n; i++) { j = i + l - 1; f[i][j] = f[i][i] + f[i + 1][j]; r[i][j] = i; for (k = i + 1; k < j; k++) if (f[i][j] < f[i][k - 1] * f[k + 1][j] + f[k][k]) { f[i][j] = f[i][k - 1] * f[k + 1][j] + f[k][k]; r[i][j] = k; } } printf("%lld\n", f[1][n]); pr(1, n); return 0; }