题意:
有n天,每天有m个小时,“1”代表有课,“0”代表没课。Ivan可以选择性地逃掉k节课,问如何逃课才能使得Ivan呆在学校的总时间最短。每天呆在学校的时间为第一节课到最后一节课的时间,例:0011010,则呆在学校的时间为4个小时。
输入:n,m,k,接着是一个n行m列的矩阵。1<=n,m<=500,0<=k<=500。
题解:
注意到数据量很小,可以想到进行三层循环都不会T,考虑dp。
定义状态:dp[i][j]表示前i天,逃了j节课,前i天呆在学校的最少时间。
还定义一个辅助状态:mn[i][j]表示第i天逃j节课,第i天呆在学校的最少时间。
则状态转移方程:dp[i][j]=min(dp[i][j],dp[i-1][j-c]+min[i][c])。其中0<=c<=j。
枚举c,c是第i天准备要逃的课数,则前i-1天逃的课数为j-c,通过这个来迭代。
mn的求法:
先预处理出每天的“1”的下标,假设存在vector中,对于第i天,枚举起始下标和终止下标,设为j和l,表示逃掉j前面的课(不包括j)和逃掉l后面的课(不包括l)。
则公式为:mn[i][vec[i].size()-l-1+j]=min(mn[i][vec[i].size()-l-1+j],vec[i][l]-vec[i][j]+1);
vec[i]表示第i天的“1”的个数,里面存的是下标。vec[i].size()-l-1+j是逃掉的课数,vec[i][l]-vec[i][j]+1则是在当前这个策略下,第i天要呆在学校的时间。
最后dp[n][k]就是答案。
#include<bits/stdc++.h>
using namespace std;
const int MAXN=510;
int n,m,k;
int dp[MAXN][MAXN],mn[MAXN][MAXN];
char str[MAXN][MAXN];
vector<int> vec[MAXN];
int main()
{
// freopen("input.txt","r",stdin);
// freopen("output.txt","w",stdout);
while(scanf("%d %d %d",&n,&m,&k)!=EOF)
{
for(int i=1;i<=n;i++) scanf("%s",&str[i][1]);
memset(mn,0x3f,sizeof(mn));
for(int i=1;i<=n;i++)
{
vec[i].clear();
for(int j=1;j<=m;j++)
{
if(str[i][j]=='1') vec[i].push_back(j);
}
for(int j=vec[i].size();j<=max(m,k);j++)
mn[i][j]=0;
}
for(int i=1;i<=n;i++)
{
for(int j=0;j<vec[i].size();j++)
{
for(int l=j;l<vec[i].size();l++)
mn[i][vec[i].size()-l-1+j]=min(mn[i][vec[i].size()-l-1+j],vec[i][l]-vec[i][j]+1);
}
}
memset(dp,0x3f,sizeof(dp));
for(int i=0;i<=k;i++) dp[0][i]=0;
for(int i=1;i<=n;i++)
{
for(int j=0;j<=k;j++)
{
for(int l=0;l<=j;l++)
dp[i][j]=min(dp[i][j],dp[i-1][j-l]+mn[i][l]);
}
}
printf("%d\n",dp[n][k]);
}
return 0;
}