Description
windy有 N 条木板需要被粉刷。每条木板被分为 M 个格子。每个格子要被刷成红色或蓝色。 windy每次粉刷,只能选择一条木板上一段连续的格子,然后涂上一种颜色。每个格子最多只能被粉刷一次。如果windy只能粉刷 T 次,他最多能正确粉刷多少格子?一个格子如果未被粉刷或者被粉刷错颜色,就算错误粉刷。
Input
输入文件paint.in第一行包含三个整数,N M T。接下来有N行,每行一个长度为M的字符串,'0'表示红色,'1'表示蓝色。
Output
输出文件paint.out包含一个整数,最多能正确粉刷的格子数。
Sample Input
3 6 3
111111
000000
001100
111111
000000
001100
Sample Output
16
HINT
30%的数据,满足 1 <= N,M <= 10 ; 0 <= T <= 100 。
100%的数据,满足 1 <= N,M <= 50 ; 0 <= T <= 2500 。
题解
类似背包的动归。
预处理:sum[i][j][0/1]表示第i块版,前j个格子有多少个颜色为0/1的。
第一步:f[i][j][k]表示第i块版,前j个格子,经过k次操作可达到的最大正确涂色个数。注意:因为一块板长度最大为50,所以最多只要50步即可涂完一整个版。
第二步:g[i][j]表示前i块版,共用了j次涂色,能得到的最大正确涂色个数。答案为g[n][p]。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m,t,sum[52][52][2],f[52][52][2502],g[52][2502];
void init()
{
char ch[55];
scanf("%d%d%d",&n,&m,&t);
for(int i=1;i<=n;i++)
{scanf("%s",ch+1);
int wc=0,bc=0;
for(int j=1;j<=m;j++)
{if(ch[j]-'0'==0) wc++;
else bc++;
sum[i][j][0]=wc; sum[i][j][1]=bc;
}
//for(int j=1;j<=m;j++) printf("%d %d\n",sum[i][j][0],sum[i][j][1]);
}
}
void dp()
{
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
for(int k=1;k<=m;k++)
for(int l=0;l<=j-1;l++)
{f[i][j][k]=max(f[i][j][k],f[i][l][k-1]+max(sum[i][j][1]-sum[i][l][1],sum[i][j][0]-sum[i][l][0]));
}
for(int i=1;i<=n;i++)
for(int j=1;j<=t;j++)
for(int k=0;k<=j;k++)
g[i][j]=max(g[i][j],g[i-1][j-k]+f[i][m][k]);
}
int main()
{
init(); dp();
printf("%d\n",g[n][t]);
}