一段序列用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;
}