Description
给出一 n×m n × m 的矩阵,初始有颜色, . . 表示白色,表示黑色,要求修改最少位置的颜色使得满足以下两个条件:
1.每列颜色相同
2.连续相同颜色的列数介于 x,y x , y 之间
Input
第一行四个整数 n,m,x,y n , m , x , y ,之后输入初始颜色状态 (1≤n,m,x,y≤1000,x≤y) ( 1 ≤ n , m , x , y ≤ 1000 , x ≤ y )
Output
输出最少需要修改颜色的位置数
Sample Input
Sample Output
11
Solution
dp[i][0/1] d p [ i ] [ 0 / 1 ] 表示前 i i 列满足条件且第列颜色为白/黑色所需要修改的最少位置,那么有转移
dp[i][0/1]=min(dp[i][0/1],dp[j][1/0]+cost(j+1,i,0/1)),x≤i−j+1≤y d p [ i ] [ 0 / 1 ] = m i n ( d p [ i ] [ 0 / 1 ] , d p [ j ] [ 1 / 0 ] + c o s t ( j + 1 , i , 0 / 1 ) ) , x ≤ i − j + 1 ≤ y
其中 cost(j+1,i,0/1) c o s t ( j + 1 , i , 0 / 1 ) 表示将第 j+1 j + 1 列到第 i i 列全部染成白/黑色所需要修改的位置数,统计每列的代价求前缀和即可得到 cost c o s t
答案即为 min(dp[m][0],dp[m][1]) m i n ( d p [ m ] [ 0 ] , d p [ m ] [ 1 ] ) ,时间复杂度 O(nm) O ( n m )
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<vector>
#include<queue>
#include<map>
#include<set>
#include<ctime>
using namespace std;
typedef long long ll;
typedef pair<int,int>P;
const int INF=0x3f3f3f3f,maxn=1005;
int n,m,x,y,dp[maxn][2],num[maxn];
char s[maxn][maxn];
int main()
{
scanf("%d%d%d%d",&n,&m,&x,&y);
for(int i=1;i<=n;i++)scanf("%s",s[i]+1);
for(int j=1;j<=m;j++)
{
for(int i=1;i<=n;i++)
if(s[i][j]!='#')num[j]++;
num[j]+=num[j-1];
}
memset(dp,INF,sizeof(dp));
for(int i=x;i<=y;i++)
dp[i][0]=n*i-num[i],dp[i][1]=num[i];
for(int i=x;i<=m;i++)
for(int j=max(0,i-y);j<=i-x;j++)
{
dp[i][1]=min(dp[i][1],dp[j][0]+num[i]-num[j]);
dp[i][0]=min(dp[i][0],dp[j][1]+(i-j)*n-(num[i]-num[j]));
}
printf("%d\n",min(dp[m][0],dp[m][1]));
return 0;
}