CodeForces 946D Timetable

Timetable

题意：ivan 是一个学生， 他们当地一周有n天， 每天有m节课，每节课一小时， 然后‘1’代表的是ivan这个时间段需要上课， 现在ivan可以跳过选择K节课不去上， 他在学校的时间为当天他选择的第一节课到当天的最后一节课，也就是可以跳2端的课， 如果当天的没有课， 当天就不需要去学校。现在求这周ivan去学校的时间最小是多少。

题解：算出不逃课的情况下的总时间， 然后用分组背包（因为每天只能有一种情况， 所以只能在每组里面选一种），跑出逃课k节下最多的逃课时间是多少， 总时间-逃课时间就是答案了。

代码：

#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define ULL unsigned LL
#define fi first
#define se second
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define max3(a,b,c) max(a,max(b,c))
const int INF = 0x3f3f3f3f;
const LL mod = 1e9+7;
typedef pair<int,int> pll;
const int N = 5e3+10;
char str[N];
int dp[N];
int pos[N];
int len[N];
int n, m, k;
int main(){
    scanf("%d%d%d", &n, &m, &k);
    int tot = 0;
    for(int i = 1; i <= n; i++) {
        scanf("%s", str+1);
        int cnt = 0;
        for(int j = 1; j <= m; j++)
            if(str[j] == '1')
                pos[++cnt] = j;
        if(cnt == 0) continue;
        memset(len, 0, sizeof(len));
        int llen = pos[cnt]-pos[1]+1;
        tot += llen;
        for(int to = 1; to <= k && to <= cnt; to++){
                for(int r = cnt - to, l = 1; r >= l && r <= cnt; r++, l++)
                    len[to] = max(len[to], llen - (pos[r] - pos[l] + 1));
                if(to == cnt)
                    len[to] = llen;
            }
        for(int j = k; j >= 1; j--)
            for(int to = 1; to <= j && to <= cnt; to++)
                dp[j] = max(dp[j], dp[j-to]+len[to]);
    }
    printf("%d", tot - dp[k]);
    return 0;
}