Codeforces 946D Timetable

题意：

有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;
}