1. 程式人生 > >【BZOJ1806】[Ioi2007]Miners 礦工配餐 DP

【BZOJ1806】[Ioi2007]Miners 礦工配餐 DP

esc 題解 順序 大寫字母 一行 比較 miners != 種類

【BZOJ1806】[Ioi2007]Miners 礦工配餐

Description

現有兩個煤礦,每個煤礦都雇用一組礦工。采煤工作很辛苦,所以礦工們需要良好飲食。每當一輛食品車到達煤礦時,礦工們便會產出一定數量的煤。有三種類型的食品車:肉車,魚車和面包車。 礦工們喜歡變化的食譜。如果提供的食品能夠不斷變化,他們的產煤量將會增加。每當一個新的食品車到達煤礦時,礦工們就會比較這種新的食品和前兩次(或者少於兩次,如果前面運送食品的次數不足兩次)的食品,並且: • 如果這幾次食品車都是同一類型的食品,則礦工們產出一個單位的煤。 • 如果這幾次食品車中有兩種不同類型的食品,則礦工們產出兩個單位的煤。 • 如果這幾次食品車中有三種不同類型的食品,則礦工們產出三個單位的煤。 預先已知食品車的類型及其被配送的順序。通過確定哪車食品送到哪個煤礦可以影響產煤量。食品車不能被拆分,每個食品車必須被全部送到一個或另一個煤礦。兩個煤礦也並不要求接收相同數量的食品車(事實上,也允許將所有食品車都送到一個煤礦)。 任務 給出食品車的類型及其被配送的順序,要求你寫一個程序,確定哪個食品車應被送到煤礦1,哪個食品車應被送到煤礦2,以使得兩個煤礦的產煤量的總和最大。

Input

輸入的第一行包含一個整數N (1 ≤ N ≤ 100 000), 表示食品車的數目。 第二行包含一個由N個字符組成的字符串,按照配送順序依次表示食品車配送的食品的類型。每個字符是以下三個大寫字母之一:‘M‘ (表示肉類), ‘F‘ (表示魚類) 或 ‘B‘ (表示面包)。

Output

輸出一個整數,表示最大的總產煤量。 評分 在45分的測試數據中,食品車的數目至多為20

Sample Input

6
MBMFFB

Sample Output

12

題解:直接用f[i][a][b][c][d]表示前i個食品車,1號礦的上一份是a,上上份是b,2號礦的上一份是c,上上份是d,的最大產量。因為卡內存,所以滾動數組即可。

#include <cstdio>
#include <cstring>
#include <iostream>

using namespace std;
int n,ans;
int f[2][4][4][4][4];
char str[100010];
int calc(int x,int a,int b)
{
	if(!a)	return 1;
	if(!b)	return 1+(x!=a);
	return 1+(x!=a||x!=b)+(x!=a&&x!=b&&a!=b);
}
int main()
{
	scanf("%d%s",&n,str+1);
	int i,j,x,a,b,c,d;
	memset(f,0x80,sizeof(f));
	f[0][0][0][0][0]=0;
	for(i=1;i<=n;i++)
	{
		j=i&1;
		memset(f[j],0x80,sizeof(f[j]));
		if(str[i]==‘M‘)	x=1;
		if(str[i]==‘F‘)	x=2;
		if(str[i]==‘B‘)	x=3;
		for(a=0;a<=3;a++)	for(b=0;b<=3;b++)	for(c=0;c<=3;c++)	for(d=0;d<=3;d++)
		{
			f[j][x][a][c][d]=max(f[j][x][a][c][d],f[j^1][a][b][c][d]+calc(x,a,b));
			f[j][a][b][x][c]=max(f[j][a][b][x][c],f[j^1][a][b][c][d]+calc(x,c,d));
		}
	}
	for(a=0;a<=3;a++)	for(b=0;b<=3;b++)	for(c=0;c<=3;c++)	for(d=0;d<=3;d++)	ans=max(ans,f[n&1][a][b][c][d]);
	printf("%d",ans);
	return 0;
}

【BZOJ1806】[Ioi2007]Miners 礦工配餐 DP