【BZOJ】作业-DP

传送门：bzoj3791-作业

---

题意

众所周知，白神是具有神奇的能力的。

比如说，他对数学作业说一声“数”，数学作业就会出于畏惧而自己完成；对语文作业说一声“语”，语文作业就会出于畏惧而自己完成。

今天，语文老师和数学老师布置了许多作业，同学们纷纷寻找白神寻求帮助。白神作为一个助人为乐的人，便答应下来。

回到家，白神将这N份作业按顺序摊开，发现语文作业数学作业混在一起，这就让白神苦恼起来，他如果对连续一段作业喊出“数”，那么里面的语文作业就会由于过于慌乱而写满错解，不过如果白神再对其喊一声“语”，它又会写满正确答案。

虽然白神很强大，但是能力还是有限制的，一天只能使用K次，现在，白神想知道他能正确的完成多少份作业。

---

数据范围

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

---

题解

我们可以发现，若可以用k次能力，那么我们最多可以完全覆盖（正确完成）的区间是：
一段0-一段1-一段0-…-一段1-一段0
最多是2*k-1段。
这样理解:每次我们选出最大范围的区间，在这个区间里选一段区间，不断的在选中的区间里选，知道用完k次，可以得出，可以分成2*k-1个区间。
设$f[i][j][0/1]$表示前i个数分成j段且当前数字为0或1（第三维表示）的最优解。
我们一波循环数组就装下了。

---

代码

#include<bits/stdc++.h>
using namespace std;
const int N=112345;
int n,k,t[N],ans,f[2][105][2];
int x,y;

inline int rd()
{
    char ch=getchar();int x=0,f=1;
    while(ch<'0' || ch>'9'){if(ch=='-') f=-1;ch=getchar();}
    while(ch>='0' && ch<='9'){x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
    return x*f;
}

int main(){
    n=rd();k=rd();
    for(int i=1;i<=n;i++) t[i]=rd();
    f[0][1][t[1]]=1;
    x=0,y=1;k<<=1;
    for(int j,a,b,i=1;i<=n-1;i++,swap(x,y)){
        for(j=1;j<k;j++){
            for(a=0;a<2;a++){
                for(b=0;b<2;b++){
                    f[y][j+(a!=b)][b]=max(f[y][j+(a!=b)][b],f[x][j][a]+(t[i+1]==b));
                }
            }
        }
        for(j=1;j<k;j++){
            f[x][j][0]=f[x][j][1]=0;
            ans=max(ans,max(f[y][j][0],f[y][j][1]));
        }
    }
    printf("%d\n",ans);
    return 0;
}