【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;
}

猜你喜欢

转载自blog.csdn.net/corsica6/article/details/80201542
今日推荐