试题 算法提高 和谐宿舍2

思路

设f[i][j]表示覆盖到i（i这个矩形已经被覆盖）的时候用了j块木板的面积最小值，然后我们枚举一下最后一块木板覆盖的矩阵是从k到i的（包括k和i矩形），转移方程：f[i][j]=min(f[i][j],f[k-1][j-1]+max*len)，其中max表示的是k到i矩形里的高度最大的值，len是i-k+1，所以我们还要预处理出来区间最大值，这里就可以用线段树来实现.

代码

#include <cstring>
#include <algorithm>
#include <iostream>
using namespace std;
const int N=10010;
const int INF=0x3f3f3f3f;
int v[N];
int f[N][N];
int m,n;
struct node{
    int l,r;
    int v;
} tr[N*4];
void build(int u,int l,int r){
    tr[u]={l,r};
    if(l==r)return;
    int mid=(l+r)>>1;
    build(u<<1,l,mid);
    build(u<<1|1,mid+1,r);
}
void modify(int u,int x,int v){
    if(tr[u].l==x&&tr[u].r==x){
        tr[u].v=v;
        return;
    }
    int mid=(tr[u].l+tr[u].r)>>1;
    if(x<=mid)modify(u<<1,x,v);
    else modify(u<<1|1,x,v);
    tr[u].v=max(tr[u<<1].v,tr[u<<1|1].v);
}
int query(int u,int l,int r){
    if(tr[u].l>=l&&tr[u].r<=r){
        return tr[u].v;
    }
    int res=0;
    int mid=(tr[u].l+tr[u].r)>>1;
    if(l<=mid)res=max(res,query(u<<1,l,r));
    if(r>mid)res=max(res,query(u<<1|1,l,r));
    return res;
}
int main(){
    cin >> n >> m;
    build(1,1,n);
    for(int i=1;i<=n;i++){
        cin >> v[i];
        modify(1,i,v[i]);
    }
    for(int i=1;i<=n;i++){
        for(int j=0;j<=m;j++){
            f[i][j]=INF;
        }
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=m;j++){
            for(int k=1;k<=i;k++){
                f[i][j]=min(f[i][j],f[k-1][j-1]+query(1,k,i)*(i-k+1));
                
            }
        }
    }
    int ans=INF;
    for(int i=1;i<=m;i++)ans=min(ans,f[n][i]);
    cout << ans;
}