[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 

题目分析

动态规划
dp[i][j1][j2][j3][j4]表示到了第i车 前两次给第一个矿地j1,j2种食物,
前两次给第二个矿地j3,j4种食物
如果j为0说明某次没有送过食物
要用滚动数组优化

#include <cstdio>
#include <cstring>
#include <cmath>
#include <set>
#include <map>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;
int n,ans;
char s[100100];
int v[100100];
int f[6][4][4][4][4];
int calc(int a,int b,int c)
{
    int tmp=1;
    if(a!=0&&a!=b&&a!=c)tmp++;
    if(b!=0&&b!=c)tmp++;
    return tmp;
}
int main()
{
    scanf("%d",&n),scanf("%s",s+1);
    for(int i=1;i<=n;i++)
        if(s[i]=='M') v[i]=1;
        else if(s[i]=='F') v[i]=2;
        else if(s[i]=='B') v[i]=3;
    memset(f,0xc0,sizeof f);
    f[0][0][0][0][0]=0;
    for(int x=1,i=1;i<=n;i++,x^=1) 
        for(int j1=0;j1<=3;j1++) 
            for(int j2=0;j2<=3;j2++) 
                for(int j3=0;j3<=3;j3++) 
                    for(int tmp,j4=0;j4<=3;j4++)
                    {
                        tmp=calc(j1,j2,v[i]);
                        f[x][v[i]][j1][j3][j4]=
                            max(f[x][v[i]][j1][j3][j4]
                                ,f[x^1][j1][j2][j3][j4]+tmp);
                        tmp=calc(j3,j4,v[i]);
                        f[x][j1][j2][v[i]][j3]=
                            max(f[x][j1][j2][v[i]][j3]
                                ,f[x^1][j1][j2][j3][j4]+tmp);
                    }
    for(int j1=0;j1<=3;j1++) 
        for(int j2=0;j2<=3;j2++) 
            for(int j3=0;j3<=3;j3++) 
                for(int j4=0;j4<=3;j4++) 
                    ans=max(ans,f[n&1][j1][j2][j3][j4]);
    printf("%d",ans);
}

发表评论

电子邮件地址不会被公开。 必填项已用*标注