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

HINT

Source

dp即可

dp[i][j][k][l][m]表示考虑第i个食物，两个矿山的前两次食物分别为jk,lm

然后直接用把i这一维滚动掉

#include<bits/stdc++.h>
 
using namespace std;
 
int n;
 
int dp[5][4][4][4][4];
 
char str[1000010];
 
int chk(char s)
{
    if(s=='M')
        return 1;
    if(s=='F')
        return 2;
    return 3;
}
 
int cal(int a,int b,int c)
{
    int cnt=1;
    if(a&&a!=b&&a!=c)
        cnt++;
    if(b&&b!=c)
        cnt++;
    return cnt;
}
 
int main()
{
    scanf("%d",&n);
    scanf("%s",str);
    memset(dp,-1,sizeof(dp));
    dp[0][0][0][0][0]=0;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<=3;j++)
        {
            for(int k=0;k<=3;k++)
            {
                for(int l=0;l<=3;l++)
                {
                    for(int m=0;m<=3;m++)
                    {
                        int now=i%4,nxt=(i+1)%4;
                        int food=chk(str[i]);
                        if(dp[now][j][k][l][m]<0) continue;
                        dp[nxt][k][food][l][m]=max(dp[nxt][k][food][l][m],dp[now][j][k][l][m]+cal(j,k,food));
                        dp[nxt][j][k][m][food]=max(dp[nxt][j][k][m][food],dp[now][j][k][l][m]+cal(l,m,food));
                    }
                }
            }
        }
    }
    int ans=-1;
    for(int i=0;i<=3;i++)
        for(int j=0;j<=3;j++)
            for(int k=0;k<=3;k++)
                for(int l=0;l<=3;l++)
                    ans=max(ans,dp[n%4][i][j][k][l]);
    printf("%d",ans);
}