BZOJ3791 作业 [DP]

一段序列用K种不同的颜色染色最多染成$2*K-1$段。。。我一开始写成$2*K$。。。貌似还与某人讨论过。。
$dp[i][j][k]$表示染到了第i个作业，共染了j段，当前染的颜色为k(k为0/1)
如果当前颜色和上一段一样，$dp[i][j][k]=max(dp[i][j][k],dp[i-1][j][k]+(a[i]==k))$;
否则如果染得不一样，也就是多染了一段，$dp[i][j][k]=max(dp[i][j][k],dp[i-1][j-1][!k]+(a[i]==k))$;
使用滚动数组优化内存。

#include<set>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100005;
void read(int &res){
    static char ch;register int flag=1;
    while((ch=getchar())<'0'||ch>'9')if(ch=='-')flag=-1;res=ch-48;
    while((ch=getchar())>='0'&&ch<='9')res=res*10+ch-48;res*=flag;
}
int dp[2][100][2],a[N],cur=1,n,m,ans;
int main(){
    read(n),read(m);
    for(register int i=1;i<=n;i++)read(a[i]);
    for(register int i=1;i<=n;i++){
        cur^=1;
        for(register int j=1;j<=m*2-1;j++){
            dp[cur][j][0]=max(dp[cur][j][0],dp[cur^1][j][0]+(a[i]==0));
            dp[cur][j][1]=max(dp[cur][j][1],dp[cur^1][j][1]+(a[i]==1));
            dp[cur][j][0]=max(dp[cur][j][0],dp[cur^1][j-1][1]+(a[i]==0));
            dp[cur][j][1]=max(dp[cur][j][1],dp[cur^1][j-1][0]+(a[i]==1));
            ans=max(ans,max(dp[cur][j][0],dp[cur][j][1]));
        }
    }
    printf("%d",ans);
    return 0;
}