蓝桥杯 算法提高 和谐宿舍2

ADV-298 和谐宿舍2

基本的动态规划问题

c++代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N = 110;

int n, m;
int h[N]; // 代表高度
int f[N][N]; // f[i][k]代表 用k块木板 从把前i个作品盖住的集合的最小值
int maxh[N][N]; // maxh[i][j]代表 i~j 个作品中 高度的最大值

int main()
{
    scanf("%d%d", &n, &m);
    int sum = 0;
    for (int i = 1; i <= n; i++) scanf("%d", &h[i]);

    // 找每个区间中最高的作品，由于数据范围很小直接暴力
    for (int i = 1; i <= n; i++)
        for (int j = i; j <= n; j++)
        {
            int s = 0;
            for (int k = i; k <= j; k++) s = max(s, h[k]);
            maxh[i][j] = s;
        }
    
    for (int i = 1; i <= n; i++)
            for (int k = 1; k <= m && k <= i; k++)
            {
                if (k == 1) f[i][k] = i * maxh[1][i]; // 只有一块木板
                else if (k == i) // 木板与作品的个数相同，则最小值为高度之和
                {
                    int sum = 0;
                    for (int u = 1; u <= i; u++) sum += h[u];
                    f[i][k] = sum;
                }
                else
                {
                    f[i][k] = 0x3f3f3f3f;
                    // 这里的u代表的是最后一块木板的起始位置
                    for (int u = i; u >= k; u--)
                        f[i][k] = min(f[i][k], f[u - 1][k - 1] + maxh[u][i] * (i - u + 1));
                }
            }

    int ans = 0x3f3f3f3f;
    for (int i = 1; i <= m; i++)
        ans = min(ans, f[n][i]);
    
    printf("%d", ans);

    return 0;
}