AcWing 479. 加分二叉树(区间 dp)

最新推荐文章于 2024-09-19 22:02:01 发布

做一只大熊猫

最新推荐文章于 2024-09-19 22:02:01 发布

阅读量289

点赞数 1

文章标签：图论算法

本文链接：https://blog.csdn.net/qq_34261446/article/details/121530269

版权

占位 94 专栏收录该内容

37 篇文章 0 订阅

订阅专栏

在这里插入图片描述

思路

一道看着很像树形 dp 的题，但其实是区间 dp。

代码

#include <bits/stdc++.h>
using namespace std;
#define db  double
#define ll  long long
#define Pir pair<int, int>
#define fi  first
#define se  second
#define pb  push_back
#define m_p make_pair
#define inf 0x3f3f3f3f
#define INF 0x3f3f3f3f3f3f3f3f
/*==========ACMer===========*/
const int N = 100;
int n, a[N];
int f[N][N];
int path[N][N];

void dfs(int l, int r) {
    if (l > r) return;
    if (l == r) {
        printf("%d ", l);
        return;
    }

    printf("%d ", path[l][r]);
    dfs(l, path[l][r] - 1);
    dfs(path[l][r] + 1, r);
}

int main()
{
    scanf("%d", &n);
    for (int i = 1; i <= n; i ++) {
        scanf("%d", &a[i]);
        f[i][i] = a[i];
    }

    for (int k = 2; k <= n; k ++) {
        for (int i = 1; i <= n - k + 1; i ++) {
            int j = i + k - 1;
            for (int md = i; md <= j; md ++) {
                int L = max(f[i][md - 1], 1);
                int R = max(f[md + 1][j], 1);
                if (f[i][j] < L * R + a[md]) {
                    f[i][j] = L * R + a[md];
                    path[i][j] = md;
                }
            }
        }
    }
    printf("%d\n", f[1][n]);
    dfs(1, n);

    return 0;
}