[洛谷]P4158 [SCOI2009]粉刷匠 (#线性dp+背包dp)

题目描述

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

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

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

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

输入格式

第一行包含三个整数，N M T。

接下来有N行，每行一个长度为M的字符串，'0'表示红色，'1'表示蓝色。

输出格式

包含一个整数，最多能正确粉刷的格子数。

输入输出样例

输入 #1复制

3 6 3
111111
000000
001100

输出 #1复制

16

说明/提示

30%的数据，满足 1 <= N,M <= 10 ； 0 <= T <= 100 。

100%的数据，满足 1 <= N,M <= 50 ； 0 <= T <= 2500 。

---

 

思路

紫题dp太毒瘤了，还是看了好长时间题解才会的......

题目要让我们在T个操作之内把一个没有涂色的木板涂成样例的样子。题目说一个格子如果未被粉刷或者被粉刷错颜色，就算错误粉刷。所以如果是颜色不对，多刷这一格不会比不刷差，就不需要考虑不刷的情况。（来自洛咕题解）

令dp[i][j][k][0/1]为在(i,j)个格子，用了k次刷子（分了k段），当前涂的颜色是0/1。

当case=color[i][j]时

dp[i][j][k][case]=max(dp[i][j-1][k][case],dp[i][j-1][k-1][case xor 1])+1

当case≠color[i][j]时

dp[i][j][k][case]=max(dp[i][j-1][k][case],dp[i][j-1][k-1][case xor 1])

处理完每一行后，max(dp[i][m][k][0]​,dp[i][m][k][1]​)即为第i行分配k次粉刷次数所能正确粉刷的最大格子数，每一行作为一组，因此在每一行跑一遍分组背包即可。

#include <stdio.h>
#include <iostream>
#define N 51
using namespace std;
int n,m,t,s,color[N][N],dp[N][N][N][2],dp_bag[2511];
signed main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	register int i,j,k,l;
	cin>>n>>m>>t;
	for(i=1;i<=n;i++)
	{
		char c[N];
		cin>>c+1;
		for(j=1;j<=m;j++)
		{
			color[i][j]=c[j]-'0';
		}
	}
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=m;j++)
		{
			for(k=1;k<=min(t,m);k++)
			{
				if(color[i][j]==0)
				{
					dp[i][j][k][0]=max(dp[i][j-1][k][0],dp[i][j-1][k-1][1])+1;
				}
				else
				{
					dp[i][j][k][0]=max(dp[i][j-1][k][0],dp[i][j-1][k-1][1]);
				}
				if(color[i][j]==1)
				{
					dp[i][j][k][1]=max(dp[i][j-1][k][1],dp[i][j-1][k-1][0])+1;
				}
				else
				{
					dp[i][j][k][1]=max(dp[i][j-1][k][1],dp[i][j-1][k-1][0]);
				}
			}
		}
	}
	for(i=1;i<=n;i++)
	{
		for(j=t;j>=0;j--)
		{
			for(k=1;k<=min(j,m);k++)
			{
				dp_bag[j]=max(dp_bag[j],dp_bag[j-k]+max(dp[i][m][k][0],dp[i][m][k][1]));
			}
		}
	}
	cout<<dp_bag[t]<<endl;
	return 0;
}