IOI2007 矿工配餐

描述

现有两个煤矿，每个煤矿都雇用一组矿工。采煤工作很辛苦，所以矿工们需要良好饮食。每当一辆食品车到达煤矿时，矿工们便会产出一定数量的煤。有三种类型的食品车：肉车，鱼车和面包车。
矿工们喜欢变化的食谱。如果提供的食品能够不断变化，他们的产煤量将会增加。每当一个新的食品车到达煤矿时，矿工们就会比较这种新的食品和前两次（或者少于两次，如果前面运送食品的次数不足两次）的食品，并且：
如果这几次食品车都是同一类型的食品，则矿工们产出一个单位的煤。
如果这几次食品车中有两种不同类型的食品，则矿工们产出两个单位的煤。
如果这几次食品车中有三种不同类型的食品，则矿工们产出三个单位的煤。
预先已知食品车的类型及其被配送的顺序。通过确定哪车食品送到哪个煤矿可以影响产煤量。食品车不能被拆分，每个食品车必须被全部送到一个或另一个煤矿。两个煤矿也并不要求接收相同数量的食品车（事实上，也允许将所有食品车都送到一个煤矿）。
任务
给出食品车的类型及其被配送的顺序，要求你写一个程序，确定哪个食品车应被送到煤矿1，哪个食品车应被送到煤矿2，以使得两个煤矿的产煤量的总和最大。

格式

输入格式

输入的第一行包含一个整数N (1 ≤ N ≤ 100 000), 表示食品车的数目。

第二行包含一个由N个字符组成的字符串，按照配送顺序依次表示食品车配送的食品的类型。每个字符是以下三个大写字母之一：’M’ (表示肉类), ‘F’ (表示鱼类) 或 ‘B’ (表示面包)。

输出格式

输出一个整数，表示最大的总产煤量。

样例1

样例输入1
6
MBMFFB

样例输出1
12

样例2

样例输入2
16
MMBMBBBBMMMMMBMB

样例输出2
29

限制

前10点时限1s,分值8分
后2点时限3s,分值10分

提示

在样例1中，可以按照如下的顺序运送食品车：煤矿 1, 煤矿 1, 煤矿 2, 煤矿 2, 煤矿 1, 煤矿 2, 依次产生的产煤量为1, 2, 1, 2, 3 和 3 个单位，一共是12 个单位。还有其它运送方式也能产生上述最大总和的产煤量。

dp题，一开始想到的是可以用二进制数来保存两个矿工当前的食品状态，后来发现没有必要，可以直接开五维数组来保存；由于 f[i] 的状态直接由 f[i-1] 推过来，所以可以用滚动数组来减小内存，当然数据范围其实可以不用滚动。

我们用 f[i][t1][t2][t3][t4] 来表示第一个矿厂前两餐是 t1 t2 ,第二个矿场前两餐是 t3 t4 ,直接枚举 t1 t2 t3 t4,往后推即可。

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
using namespace std;
int f[100001][4][4][4][4],dui[100001];
int cal(int a,int b,int c)
{
    int x=1;
    if(a!=0&&a!=b&&a!=c)x++;
    if(b!=0&&b!=c)x++;
    return x;
}
int main()
{
    int n;
    cin>>n;
    getchar();
    memset(f,128,sizeof(f));
    for(int i=1;i<=n;i++)
    {
        char c;
        scanf("%c",&c);
        if(c=='M')
            dui[i]=1;
        else if(c=='F')
            dui[i]=2;
        else dui[i]=3;
    }
    f[0][0][0][0][0]=0;
    //int now=0,pre=1;滚动
    for(int i=1;i<=n;i++)
    {
        for(int t1=0;t1<4;t1++)
            for(int t2=0;t2<4;t2++)
                for(int t3=0;t3<4;t3++)
                    for(int t4=0;t4<4;t4++)
                    {
                        if(0>/**/f[i-1][t1][t2][t3][t4])//之前的状态必须推过才能转移到现在这个状态
                            continue;
                        int add=cal(t1,t2,dui[i]);
                        f[i][t2][dui[i]][t3][t4]=max(f[i][t2][dui[i]][t3][t4],f[i-1][t1][t2][t3][t4]+add);//餐车分配给第一个矿场
                        add=cal(t3,t4,dui[i]);
                        f[i][t1][t2][t4][dui[i]]=max(f[i][t1][t2][t4][dui[i]],f[i-1][t1][t2][t3][t4]+add);//餐车分配给第二个矿场
                    }
    }
    int ans=0;
    for(int t1=0;t1<4;t1++)
        for(int t2=0;t2<4;t2++)
            for(int t3=0;t3<4;t3++)
                for(int t4=0;t4<4;t4++)
                    ans=max(ans,f[n][t1][t2][t3][t4]);
    printf("%d",ans);
}