C Rabbit的工作(1)

转载出处：https://blog.nowcoder.net/n/0134a33f23e2401285d56d7d5c15af56?&toCommentId=6656404
题目链接
很明显的DP了，但是这个题有意思的一点是要用滚动数组，不然会MLE。
首先看到n是400，400400400也不到1e8，可以用n^3的暴力。
所以联想到dp数组要开三维，正常来说是这三维：dp[i][j][k]
i表示前i天，j是已经工作了j天，k是到目前已连续工作了多少天。
那么很容易得到转移方程：
如果这天不工作或者这天本来就不用工作：dp[i][j][0]=min(dp[i][j][0],dp[i-1][j][k])
如果今天需要工作：dp[i][j][k]=min(dp[i][j][k],dp[i-1][j-1][k-1]+k)
但是因为会MLE，同时我们发现dp数组只跟dp[i-1]有关，所以我们可以把第一维压到2，通过奇偶来表示不同的上下两层就好了。

#include<cstdio>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<cstring>
#include<queue>
#define fs first
#define se second
#define pb push_back
#define cppio ios::sync_with_stdio(false);cin.tie(0)
#define eps 1e-7
using namespace std;
 
typedef long long ll;
typedef pair<int,int> pii;
typedef vector<int> VI;
 
const int maxn=405;
const ll inf=0x3f3f3f3f;
const ll mod=1e9+7;
 
char s[maxn];
 
int dp[2][maxn][maxn];
 
int main(){
    int n,K;
    scanf("%d%d%s",&n,&K,s+1);
    memset(dp,0x3f,sizeof(dp));
    dp[0][0][0]=0;
    for(int i=1;i<=n;i++){
        memset(dp[i%2],0x3f,sizeof(dp[i%2]));
        for(int j=i;j>=0;j--){
            for(int k=0;k<=j;k++){
                dp[i%2][j][0]=min(dp[i%2][j][0],dp[(i+1)%2][j][k]);
                if(s[i]=='1'&&k>0){
                    dp[i%2][j][k]=min(dp[i%2][j][k],dp[(i+1)%2][j-1][k-1]+k);
                }
            }
        }
    }
    for(int i=n;i>=0;i--){
        for(int j=i;j>=0;j--){
            if(dp[n%2][i][j]<=K){
                printf("%d\n",i);return 0;
            }
        }
    }
    return 0;
}