bzoj3791作业*

bzoj3791作业

题意：

对一个01序列进行染色，每次能将一个区间染上色（可覆盖之前染的），共能染k次，求最大正确染色个数。n≤100000，m≤50。

题解：

结论：染k次最多能把序列分成2*k-1段。故dp即可：

f[i][j][0]=max(f[i+1][j+1][1]+a[i]==1,f[i+1][j][0]+a[i]==0)

f[i][j][1]=max(f[i+1][j][1]+a[i]==1,f[i+1][j+1][0]+a[i]==0)

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#define inc(i,j,k) for(int i=j;i<=k;i++)
#define maxn 110
#define INF 0x3fffffff
using namespace std;

inline int read(){
    char ch=getchar(); int f=1,x=0;
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return f*x;
}
int n,f[2][maxn][3],k,x,y; bool a[maxn*1000];
int main(){
    n=read(); k=2*read()-1; inc(i,1,n)a[i]=read(); x=0; y=1;
    for(int i=n;i>=1;i--){
        f[x][k+1][0]=f[x][k+1][1]=-INF;
        inc(j,1,k){
            f[y][j][0]=max(f[x][j][0]+(a[i]==0),f[x][j+1][1]+(a[i]==1));
            f[y][j][1]=max(f[x][j+1][0]+(a[i]==0),f[x][j][1]+(a[i]==1));
        }
        swap(x,y);
    }
    printf("%d",max(f[x][1][0],f[x][1][1])); return 0;
}

20160831