【bzoj3791】作业

Description

众所周知，白神是具有神奇的能力的。
比如说，他对数学作业说一声“数”，数学作业就会出于畏惧而自己完成；对语文作业说一声“语”，语文作业就会出于畏惧而自己完成。
今天，语文老师和数学老师布置了许多作业，同学们纷纷寻找白神寻求帮助。白神作为一个助人为乐的人，便答应下来。
回到家，白神将这N份作业按顺序摊开，发现语文作业数学作业混在一起，这就让白神苦恼起来，他如果对连续一段作业喊出“数”，那么里面的语文作业就会由于过于慌乱而写满错解，不过如果白神再对其喊一声“语”，它又会写满正确答案。
虽然白神很强大，但是能力还是有限制的，一天只能使用K次，现在，白神想知道他能正确的完成多少份作业。

Input

第一行两个整数N，K。
第二行N个0或者1表示这份作业是语文作业还是数学作业。

Output

输出一个整数，表示白神能正确完成的作业数。

Sample Input

5 2
0 1 0 1 0
Sample Output

4
HINT

100%的数据中N ≤ 100000，K<=50.

题解
k次相当于最多分成2*k-1段，每段为0或1，使得覆盖最多，转移很简单。

代码

#include<bits/stdc++.h>
#define inf 1000000000
#define ll long long
using namespace std;
inline int read()
{
    int x=0;char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
int n,k,a[100005],dp[2][105][2],ans,x;
int main()
{
    n=read();k=read();
    for (int i=1;i<=n;i++) a[i]=read();
    for (int i=1;i<=n;i++)
    {
        x=x^1;memset(dp[x],0,sizeof(dp[x]));
        for (int j=1;j<=2*k-1;j++)
        {
            if (j>i) break;
            dp[x][j][0]=dp[x^1][j-1][1]+(a[i]==0);
            dp[x][j][0]=max(dp[x][j][0],dp[x^1][j][0]+(a[i]==0));
            dp[x][j][1]=dp[x^1][j-1][0]+(a[i]==1);
            dp[x][j][1]=max(dp[x][j][1],dp[x^1][j][1]+(a[i]==1));
        }
    }
    for (int i=1;i<=2*k-1;i++)
        ans=max(ans,max(dp[x][i][0],dp[x][i][1]));
    printf("%d",ans);
    return 0;
}