bzoj1806: [Ioi2007]Miners 矿工配餐 DP

Description
现有两个煤矿，每个煤矿都雇用一组矿工。采煤工作很辛苦，所以矿工们需要良好饮食。每当一辆食品车到达煤矿时，矿工们便会产出一定数量的煤。有三种类型的食品车：肉车，鱼车和面包车。 矿工们喜欢变化的食谱。如果提供的食品能够不断变化，他们的产煤量将会增加。每当一个新的食品车到达煤矿时，矿工们就会比较这种新的食品和前两次（或者少于两次，如果前面运送食品的次数不足两次）的食品，并且： • 如果这几次食品车都是同一类型的食品，则矿工们产出一个单位的煤。 • 如果这几次食品车中有两种不同类型的食品，则矿工们产出两个单位的煤。 • 如果这几次食品车中有三种不同类型的食品，则矿工们产出三个单位的煤。 预先已知食品车的类型及其被配送的顺序。通过确定哪车食品送到哪个煤矿可以影响产煤量。食品车不能被拆分，每个食品车必须被全部送到一个或另一个煤矿。两个煤矿也并不要求接收相同数量的食品车（事实上，也允许将所有食品车都送到一个煤矿）。 任务 给出食品车的类型及其被配送的顺序，要求你写一个程序，确定哪个食品车应被送到煤矿1，哪个食品车应被送到煤矿2，以使得两个煤矿的产煤量的总和最大。
Input
输入的第一行包含一个整数N (1 ≤ N ≤ 100 000), 表示食品车的数目。 第二行包含一个由N个字符组成的字符串，按照配送顺序依次表示食品车配送的食品的类型。每个字符是以下三个大写字母之一：’M’ (表示肉类), ‘F’ (表示鱼类) 或 ‘B’ (表示面包)。
Output
输出一个整数，表示最大的总产煤量。

直接记录两个矿场的情况，注意非法情况赋值为-inf
转移时i-1天变成i-2天，注意细节。
这种简单的题应该一次AC！！！

#include<bits/stdc++.h>
using namespace std;
#define maxn 100020
#define inf 0x3f3f3f3f

int f[2][4][4][4][4];
char ch[maxn];
int a[maxn],n,ans;

inline int cal(int x,int y,int z){
    if ( !x ){
        if ( !y ) return 1;
        return (y == z) ? 1 : 2;
    }
    if ( !y ) return (x == z) ? 1 : 2;
    if ( (x == y) && (y == z) ) return 1;
    if ( (x == y) || (y == z) || (x == z) ) return 2;
    return 3;
}
void init(int t){
    memset(f[t],-1,sizeof(f[t]));
/*  for (int j = 0 ; j < 4 ; j++)
        for (int k = 0 ; k < 4 ; k++)
            for (int x = 0 ; x < 4 ; x++)
                for (int y = 0 ; y < 4 ; y++)
                    f[t][j][k][x][y] = -inf;*/
}
int main(){
    freopen("input.txt","r",stdin);
//  freopen("1.out","w",stdout);
    scanf("%d",&n);
    scanf("%s",ch + 1);
    for (int i = 1 ; i <= n ; i++) 
        if ( ch[i] == 'M' ) a[i] = 3;
        else if ( ch[i] == 'B' ) a[i] = 1;
        else a[i] = 2;
    int t = 0;
    init(t) , init(t ^ 1) , f[t][0][0][0][0] = 0;
    for (int i = 1 ; i <= n ; i++,t ^= 1,init(t ^ 1))
        for (int j = 0 ; j < 4 ; j++)
            for (int k = 0 ; k < 4 ; k++)
                for (int x = 0 ; x < 4 ; x++)
                    for (int y = 0 ; y < 4 ; y++){
                        if ( f[t][j][k][x][y] < 0 ) continue;
                        //cout<<i<<" "<<j<<" "<<k<<" "<<x<<" "<<y<<" "<<f[t][j][k][x][y]<<endl;
                        f[t ^ 1][k][a[i]][x][y] = max(f[t ^ 1][k][a[i]][x][y],f[t][j][k][x][y] + cal(j,k,a[i]));
                        f[t ^ 1][j][k][y][a[i]] = max(f[t ^ 1][j][k][y][a[i]],f[t][j][k][x][y] + cal(x,y,a[i]));
                    }
    for (int j = 0 ; j < 4 ; j++)
        for (int k = 0 ; k < 4 ; k++)
            for (int x = 0 ; x < 4 ; x++)
                for (int y = 0 ; y < 4 ; y++)
                    ans = max(f[t][j][k][x][y],ans);
    cout<<ans<<endl;
    return 0;
}