bzoj 1296: [SCOI2009]粉刷匠 动态规划

题意

windy有 N 条木板需要被粉刷。 每条木板被分为 M 个格子。 每个格子要被刷成红色或蓝色。 windy每次粉刷，只能选择一条木板上一段连续的格子，然后涂上一种颜色。 每个格子最多只能被粉刷一次。 如果windy只能粉刷 T 次，他最多能正确粉刷多少格子？ 一个格子如果未被粉刷或者被粉刷错颜色，就算错误粉刷。
1 <= N,M <= 50 ； 0 <= T <= 2500 。

分析

设f[i,j,k]表示按从左到右，从上到下的顺序，刷到了第i行第j列，刷了j次的最多正确格子。
假设a[i,j]=1，不难注意到f[i,j,k]只会由f[i,l,k-1]转移过来，其中满足f[i,l]=0。a[i,j]=0时同理。
那么我们就可以用类似单调队列优化多重背包的方法来优化转移即可。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

const int N=55;
const int inf=100000000;

int n,m,T,f[N][N][N*N],a[N][N];
char ch[N];

int main()
{
    scanf("%d%d%d",&n,&m,&T);
    for (int i=1;i<=n;i++)
    {
        scanf("%s",ch+1);
        for (int j=1;j<=m;j++) a[i][j]=ch[j]-'0';
    }
    int tag0=0,tag1=0,mx1=0,mx0=0;
    for (int k=1;k<=T;k++)
        for (int i=1;i<=n;i++)
            for (int j=1;j<=m;j++)
            {
                int x=i,y=j-1;
                if (!y) x--,y=m,tag0=tag1=0,mx0=mx1=f[x][y][k-1];
                f[i][j][k]=f[x][y][k];
                if (!a[i][j]) f[i][j][k]=max(f[i][j][k],mx1+tag1+1),tag1++,mx0=max(mx0,f[i][j][k-1]-tag0);
                else f[i][j][k]=max(f[i][j][k],mx0+tag0+1),tag0++,mx1=max(mx1,f[i][j][k-1]-tag1);
            }
    printf("%d",f[n][m][T]);
    return 0;
}