【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.

Source

题意转化:
一个序列已知
另一个序列做k次区间染色,问最多有多少元素与已知序列相同

染色显然最多染成2k-1段
直接DP
f[i][j][k]表示前i个元素,分j段,最后一个元素颜色为k最多有多少

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define MAXN 100010
#define GET (ch>='0'&&ch<='9')
using namespace std;
int n,k,ans;
int a[MAXN];
int f[MAXN][110][2];
void in(int &x)
{
    char ch=getchar();x=0;
    while (!GET)    ch=getchar();
    while (GET) x=x*10+ch-'0',ch=getchar();
}
int main()
{
    in(n);in(k);
    for (int i=1;i<=n;i++)  in(a[i]);f[1][1][a[1]]=1;
    for (int j=1;j<=(k<<1)-1;j++)
    {
        for (int i=1;i<=n;i++)  for (int k=0;k<=1;k++)
            f[i][j][k]=max(f[i-1][j-1][k^1],f[i-1][j][k])+(a[i]==k),
            ans=max(ans,f[i][j][k]);
    }
    printf("%d\n",ans);
}