bzoj1806 [Ioi2007]Miners 矿工配餐

Description

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

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

Solution

---

状态还是很好想的，f[I,j,k]表示前i个车，送去1矿的最后两个状态为j，送去2矿的最后两个状态为k的最大产煤量，转移的时候记得判断状态的合法性

Code

---

#include <stdio.h>
#include <string.h>
#include <algorithm>
#define rep(i,st,ed) for (int i=st;i<=ed;++i)
#define fill(x,t) memset(x,t,sizeof(x))

const int N=200005;

int f[2][16][16];
int a[N];

char str[N];

int cal(int x,int k) {
    int rec[4]={0,0,0,0},ret=0;
    if (x%4) ret+=!(rec[x%4]++);
    if (x/4) ret+=!(rec[x/4]++);
    ret+=!rec[k];
    return ret;
}

int main(void) {
    int n; scanf("%d",&n);
    scanf("%s",str+1);
    rep(i,1,n) {
        if (str[i]=='M') a[i]=1;
        if (str[i]=='F') a[i]=2;
        if (str[i]=='B') a[i]=3;
    }
    f[0][0][0]=1;
    rep(i,0,n-1) {
        fill(f[(i+1)&1],0);
        int tot;
        rep(j,0,15) rep(k,0,15) {
            if (!f[i&1][j][k]) continue;
            tot=cal(j,a[i+1]);
            f[(i+1)&1][j%4*4+a[i+1]][k]=std:: max(f[(i+1)&1][j%4*4+a[i+1]][k],f[i&1][j][k]+tot);
            tot=cal(k,a[i+1]);
            f[(i+1)&1][j][k%4*4+a[i+1]]=std:: max(f[(i+1)&1][j][k%4*4+a[i+1]],f[i&1][j][k]+tot);
        }
    }
    int ans=0;
    rep(i,0,15) rep(j,0,15) ans=std:: max(ans,f[n&1][i][j]);
    printf("%d\n", ans-1);
    return 0;
}