牛客 粉刷匠（dp）

有些题目可以进行二维dp，当然这题用四维也可以做。

我们先做每一行，f[][][]，表示第i个用j次，涂前k个的最大值。

做完后，可以把它看作分组背包问题，每个木板都是一个物品，再跑一边dp即可

#include<bits/stdc++.h>
using namespace std;
const int N=3e5+10;
int f[55][2510][55];
int g[55][2510];
string s[N];
int w[55][55][55];
int main(){
    int n,m,t;
    cin>>n>>m>>t;
    int i,j,k;
    for(i=1;i<=n;i++){
        cin>>s[i];
        s[i]=" "+s[i];
        for(j=1;j<=m;j++){
            int cnt1=0,cnt2=0;
            for(k=j;k>=1;k--){
                if(s[i][k]=='1')
                    cnt1++;
                else
                    cnt2++;
                w[i][k][j]=max(cnt1,cnt2);
            }
        }
    }
    for(k=1;k<=n;k++){
        for(i=1;i<=m;i++){
            for(j=1;j<=m;j++){
                for(int l=0;l<j;l++){
                    f[k][i][j]=max(f[k][i][j],f[k][i-1][l]+w[k][l+1][j]);
                }
            }
        }
    }
    int ans=0;
    for(i=1;i<=n;i++){
        for(j=1;j<=t;j++){
            for(k=0;k<=min(j,m);k++)
                g[i][j]=max(g[i][j],g[i-1][j-k]+f[i][k][m]);
            ans=max(g[i][j],ans);
        }
    }
    cout<<ans<<endl;
}

View Code