1. 程式人生 > >洛谷1040 加分二叉樹

洛谷1040 加分二叉樹

原題連結

挺水的一道區間\(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]\)

表示在中序遍歷下編號\(i \sim j\)的點所構成的子樹的最高加分所選擇的根,在\(DP\)過程中不斷更新,最後遞迴輸出即可。

#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;
}