首页 > 代码库 > BZOJ 1806 IOI2007 Miners 矿工配餐 动态规划

BZOJ 1806 IOI2007 Miners 矿工配餐 动态规划

题目大意:将一个123序列拆分为两个子序列,定义每个数的贡献值为以这个数结尾的长度最大为3的子串中不同数的数量,求贡献值和的最大值

令f[i][a1][a2][b1][b2]为前i个数分成两组,第一组以a1 a2结尾,第二组以b1 b2结尾的最大贡献值 转移啥的自己YY吧 记得开滚动数组

尼玛写错个参数都要调半天……

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,ans,f[2][4][4][4][4];
inline int Get_Char()
{
	char c;
	do c=getchar(); while(c!='M'&&c!='F'&&c!='B');
	switch(c)
	{
		case 'M':return 1;
		case 'F':return 2;
		case 'B':return 3;
	}
}
inline int Score(int x,int y)
{
	if(x==0) return 1;
	return x==y?1:2;
}
inline int Score(int x,int y,int z)
{
	int temp=x*y*z;
	switch(temp)
	{
		case 6:return 3;
		case 1:
		case 8:
		case 27:return 1;
		case 0:return Score(y,z);
		default: return 2;
	}
}
int main()
{
	int i,a1,a2,b1,b2;
	cin>>n;
	memset(f,0xef,sizeof f);
	f[0][0][0][0][0]=0;
	for(i=1;i<=n;i++)
	{
		int temp=Get_Char();
		memset(f+(i&1),0xef,sizeof(f)>>1);
		for(a1=0;a1<=3;a1++)
			for(a2=(a1?1:0);a2<=3;a2++)
				for(b1=0;b1<=3;b1++)
					for(b2=(b1?1:0);b2<=3;b2++)
						if(f[~i&1][a1][a2][b1][b2]>=0)
						{
							f[i&1][a2][temp][b1][b2]=max(f[i&1][a2][temp][b1][b2]
														,f[~i&1][a1][a2][b1][b2]+Score(a1,a2,temp));
							f[i&1][a1][a2][b2][temp]=max(f[i&1][a1][a2][b2][temp]
														,f[~i&1][a1][a2][b1][b2]+Score(b1,b2,temp));
						}
	}
	for(a1=0;a1<=3;a1++)
		for(a2=(a1?1:0);a2<=3;a2++)
			for(b1=0;b1<=3;b1++)
				for(b2=(b1?1:0);b2<=3;b2++)
					ans=max(ans,f[n&1][a1][a2][b1][b2]);
	cout<<ans<<endl;
}


BZOJ 1806 IOI2007 Miners 矿工配餐 动态规划