【bzoj3791】作业 dp

最后整个序列最多被分为2K-1段黑白相间，因为每次等价于插入2个端点
换种说法，只要是被分成不超过2K-1个黑白相间的段，那么就可以用最多K次解决
dp[i][j][0/1]表示前i个数，用了j段最后一段为黑还是白的总数

有点厉害

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>

using namespace std;

int f[100010][100][2];
int n,m,k;

int main()
{
	scanf("%d%d",&n,&k);
	memset(f,-1,sizeof(f));
	f[0][0][0]=f[0][0][1]=0;
	for (int i=1;i<=n;i++)
	{
		int x;
		scanf("%d",&x);
		for (int j=1;j<=min(2*k-1,i);j++)
		  for (int k=0;k<2;k++)
		  {
		    f[i][j][k]=f[i-1][j][k]+(x==k);
		    for (int p=0;p<2;p++)
		      f[i][j][k]=max(f[i][j][k],f[i-1][j-1][p]+(x==k));
		  }
	}
	int ans=0;
	for (int i=1;i<=2*k-1;i++) ans=max(ans,max(f[n][i][0],f[n][i][1]));
	printf("%d\n",ans);
	return 0;
}