Description
windy有 N 条木板需要被粉刷。
每条木板被分为 M 个格子。
每个格子要被刷成红色或蓝色。
windy每次粉刷,只能选择一条木板上一段连续的格子,然后涂上一种颜色。
每个格子最多只能被粉刷一次。
如果windy只能粉刷 T 次,他最多能正确粉刷多少格子?
一个格子如果未被粉刷或者被粉刷错颜色,就算错误粉刷。
Input
第一行包含三个整数,N M T。
接下来有N行,每行一个长度为M的字符串,'0’表示红色,'1’表示蓝色。
Output
输出一个整数,表示最多能正确粉刷的格子数。
Sample Input
3 6 3
111111
000000
001100
Sample Output
16
Data Constraint
Hint
100%的数据,满足 1 <= N,M <= 50 ; 0 <= T <= 2500 。
.
.
.
.
.
.
分析
定义一个f[i][j][k]来表示第i条木板上粉刷j次涂了前k个格子的情况下能正确粉刷的最大格子数
用sum数组来记录蓝色格子数 某个区间的格子数减去蓝色格子数就是红色格子数(用前缀和来记录)
接下来就是求f[i][j][k]态转移方程了还是很好想的
前l个格子粉刷正确加上下一步粉刷正确的红色格子多还是蓝色格子多
有f[i][j][k]=max(f[i][j][k],f[i][j-1][l]+max(sum[i][k]-sum[i][l],k-l-sum[i][k]+sum[i][l]));
其中sum[i][k]-sum[i][l]为蓝色格子数 k-l-sum[i][k]+sum[i][l]为该段的红色格子数
定义一个 a[i][j]的数组来表示前i条木板粉刷j次的情况下能正确粉刷的最大格子数
要是求最大的格子数 那么就要求一个m使得
a[i][j]=max(a[i][j],a[i-1][j-k]+f[i][k][m])
最后看看粉刷多少次时有最大的a[i][j]记作ans;
.
.
.
.
.
.
程序:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int n,m,t,a[60][2550],sum[60][2510],f[60][2510][60],ans=0;
int main()
{
scanf("%d%d%d",&n,&m,&t);
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++)
{
char x;
cin>>x;
if (x=='1') sum[i][j]=sum[i][j-1]+1; else sum[i][j]=sum[i][j-1];
}
for (int i=1;i<=n;i++)
for (int j=1;j<=m;j++)
for (int k=1;k<=m;k++)
for (int l=j-1;l<=k-1;l++)
f[i][j][k]=max(f[i][j][k],f[i][j-1][l]+max(sum[i][k]-sum[i][l],k-l-(sum[i][k]-sum[i][l])));
for (int i=1;i<=n;i++)
for (int j=1;j<=t;j++)
for (int k=0;k<=min(j,m);k++)
a[i][j]=max(a[i][j],a[i-1][j-k]+f[i][k][m]);
for (int i=1;i<=t;i++)
ans=max(ans,a[n][i]);
printf("%d",ans);
return 0;
}