[SCOI2009]粉刷匠 两个dp

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网

题目描述

windy有 N 条木板需要被粉刷。 每条木板被分为 M 个格子。 每个格子要被刷成红色或蓝色。 

windy每次粉刷，只能选择一条木板上一段连续的格子，然后涂上一种颜色。 每个格子最多只能被粉刷一次。 

如果windy只能粉刷 T 次，他最多能正确粉刷多少格子？ 

一个格子如果未被粉刷或者被粉刷错颜色，就算错误粉刷。

输入描述:

输入文件paint.in第一行包含三个整数，N M T。
接下来有N行，每行一个长度为M的字符串，'0'表示红色，'1'表示蓝色。

输出描述:

输出文件paint.out包含一个整数，最多能正确粉刷的格子数。

示例1

输入

3 6 3
111111
000000
001100

输出

16

考虑用动态规划,这里可以发现每个木板实际上是互不干扰的，所以可以先把一个大问题分解一下，可以先去算出每个木板在z(1---t)次粉刷情况下最多能达到多少正确，再去把这些情况逐个对应到总的情况里面去看看有没有更优。

先看第一个问题，要算出每个木板在粉刷x次的情况下最多有多少个正确，设dp[i][x]为木板的前j个块粉刷x次最多有多少正确，再看第j个块，他此时只有两种状态，要么粉刷，要么不粉刷，不粉刷简单一点，承接前面dp[j-1][x]即可，如果要粉刷的话，那么这个块要么自己一块，要么跟前面一个，两个...于是方程可以确定为dp[k][x-1]+num,这里num指的是k+1到j最多得多少正确，实际上就是统计这个区间0和1谁多，加上.

       for(int t=1;t<=m;t++)
            sum[t]=sum[t-1]+(a[t]=='1');//计算到这个位置的1的数量
	    for(int x=1;x<=t;x++)//粉刷的次数
			for (int j = 1; j <= m; j++)
			{
				dp[j][x] =dp[j-1][x];//不粉刷这个，先承接上一次的
				//枚举粉刷断点
				for (int k = j-1; k >= 0; k--)
				{
					dp[j][x] = max(dp[j][x] 
                               ,dp[k][x - 1] + max(sum[j]-sum[k],(j-k)-(sum[j]-sum[k])));
				}
			}

第二个问题就是汇总的问题，枚举每个单个粉刷量和对应的正确数，经典0/1背包,这里数据量不大，直接写

//dp[m][k]此时每个木板表示取k个的最多正确数
		for (int j = t; j >= 1; j--)//背包
			for (int k = 1; k <= j; k++)//粉刷次数
				f[j] = max(f[j], f[j - k] + dp[m][k]);

完整代码段

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
char a[100];
int dp[100][5000];//dp[i][x]表示到第i个数字刷x次可以达到的最大值
int f[5000];//背包
int sum[1000];//计算到这个位置有几个1
int n, m, t;
int main()
{
	scanf("%d%d%d", &n, &m, &t);
	for (int i = 1; i <= n; i++)//计算每个木板刷x次可以得到的最大收益再装进背包
	{
		scanf("%s", a +1);
        for(int t=1;t<=m;t++)
            sum[t]=sum[t-1]+(a[t]=='1');//计算到这个位置的1的数量
	    for(int x=1;x<=t;x++)//粉刷的次数
			for (int j = 1; j <= m; j++)
			{
				dp[j][x] =dp[j-1][x];//不粉刷这个，先承接上一次的
				//枚举粉刷断点
				for (int k = j-1; k >= 0; k--)
				{
					dp[j][x] = max(dp[j][x] 
                              ,dp[k][x - 1] + max(sum[j]-sum[k],(j-k)-(sum[j]-sum[k])));
				}
			}
        //dp[m][k]此时每个木板表示取k个的最多正确数
		for (int j = t; j >= 1; j--)//背包
			for (int k = 1; k <= j; k++)//粉刷次数
				f[j] = max(f[j], f[j - k] + dp[m][k]);
	}
	printf("%d\n", f[t]);
	return 0;
}