1. 程式人生 > >codeforces 946 D Timetable(預處理+分組揹包)

codeforces 946 D Timetable(預處理+分組揹包)

思路:先預處理出每天去掉x節課能少去學校的時間,然後就變成了n組每組挑一個,容量大小為K的分組揹包了。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <cctype>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <sstream>
#include<iomanip>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll inff = 0x3f3f3f3f3f3f3f3f;
#define FOR(i,a,b) for(int i(a);i<=(b);++i)
#define FOL(i,a,b) for(int i(a);i>=(b);--i)
#define REW(a,b) memset(a,b,sizeof(a))
#define inf int(0x3f3f3f3f)
#define si(a) scanf("%d",&a)
#define sl(a) scanf("%I64d",&a)
#define sd(a) scanf("%lf",&a)
#define ss(a) scanf("%s",a)
#define mod int(1e9+7)
#define lc (d<<1)
#define rc (d<<1|1)
#define P pair<int,int>
#define pi acos(-1)
int n,m,k;
string a;
int b[508],c[508][508],d[508][508],dp[508];
int main()
{
    cin.tie(0);
    si(n),si(m),si(k);
    REW(b,0),REW(c,0),REW(d,0),REW(dp,0);
    int s=0;
    FOR(i,1,n)
    {
        cin>>a;
        int z=0;
        FOR(j,0,m-1)
        {
            if(a[j]=='1')  c[i][++z]=j;
        }
        c[i][0]=z;
        FOR(j,1,min(z-1,k))
        {
            FOR(f,0,j)
            {
                d[i][j]=max(d[i][j],c[i][z]-c[i][1]-c[i][z-j+f]+c[i][f+1]);
            }
        }
        if(z)   s+=(c[i][z]-c[i][1]+1),d[i][z]=(c[i][z]-c[i][1]+1);;
    }
    FOR(i,1,n)
    {
        FOL(j,k,0)
        {
            FOR(f,0,min(k,c[i][0]))
            {
                if(j-f>=0)  dp[j]=max(dp[j],dp[j-f]+d[i][f]);
            }
        }
    }
    cout<<s-dp[k]<<endl;
    return 0;
}