[Luogu P4401] [BZOJ 1806] [IOI2007]Miners 矿工配餐

题目描述

现有两个煤矿，每个煤矿都雇用一组矿工。采煤工作很辛苦，所以矿工们需要良好饮食。每当一辆食品车到达煤矿时，矿工们便会产出一定数量的煤。

有三种类型的食品车：肉车，鱼车和面包车。 矿工们喜欢变化的食谱。如果提供的食品能够不断变化，他们的产煤量将会增加。每当一个新的食品车到达煤矿时，矿工们就会比较这种新的食品和前两次（或者少于两次，如果前面运送食品的次数不足两次）的食品，并且：

• 如果这几次食品车都是同一类型的食品，则矿工们产出一个单位的煤。
• 如果这几次食品车中有两种不同类型的食品，则矿工们产出两个单位的煤。
• 如果这几次食品车中有三种不同类型的食品，则矿工们产出三个单位的煤。

预先已知食品车的类型及其被配送的顺序。通过确定哪车食品送到哪个煤矿可以影响产煤量。食品车不能被拆分，每个食品车必须被全部送到一个或另一个煤矿。两个煤矿也并不要求接收相同数量的食品车（事实上，也允许将所有食品车都送到一个煤矿）。

给出食品车的类型及其被配送的顺序，要求你写一个程序，确定哪个食品车应被送到煤矿1，哪个食品车应被送到煤矿2，以使得两个煤矿的产煤量的总和最大。

输入输出格式

输入格式：

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

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

输出格式：

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

输入输出样例

输入样例#1：

6
MBMFFB

输出样例#1：

12

输入样例#2：

16
MMBMBBBBMMMMMBMB

输出样例#2：

29

说明

在样例1中，可以按照如下的顺序运送食品车：煤矿 1, 煤矿 1, 煤矿 2, 煤矿 2, 煤矿 1, 煤矿 2, 依次产生的产煤量为$1, 2, 1, 2, 3$ 和 $3$ 个单位，一共是$12$ 个单位。

还有其它运送方式也能产生上述最大总和的产煤量。

解题分析

简单dp。设$dp[now][a1][a2][b1][b2]$表示当前考虑到第$now$个食物，类型为$typ$，煤矿1之前两个得到的食物分别是$a1,a2$，煤矿2之前得到的食物分别是$b1,b2$， 则有如下转移方程：

$$dp[now][a1][a2][b1][b2]=max(dp[now+1][a2][typ][b1][b2]+val(a1,a2,typ),dp[now+1][a1][a2][b2][typ]+val(b1,b2,typ))$$
一开始博主为了图方便打了记忆化搜索，结果是这样的：

才注意到内存只有18M…不过仍然把代码丢出来：

#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#define R register
#define IN inline
#define W while
#define MX 100050
char buf[MX];
int typ[MX], len, dp[MX][4][4][4][4];
IN int get(const int &g1, const int &g2, const int &now)
{
    if(!g1 && !g2) return 1;
    if(!g1) return ((g2 != now) + 1);
    if(g1 == g2 && g2 == now) return 1;
    return (g1 != g2) + (g2 != now) + (g1 != now);
}
int DFS(const int &now, const int &a1, const int &a2, const int &b1, const int &b2)
{
    int &tar = dp[now][a1][a2][b1][b2];
    if(now > len) return 0;
    if(tar) return tar;
    return tar = std::max(DFS(now + 1, a2, typ[now], b1, b2) + get(a1, a2, typ[now]), DFS(now + 1, a1, a2, b2, typ[now]) + get(b1, b2, typ[now]));
}
int main(void)
{
    scanf("%d%s", &len, buf + 1);
    for (R int i = 1; i <= len; ++i)
    {
        switch (buf[i])
        {
            case 'M': typ[i] = 1; break;
            case 'F': typ[i] = 2; break;
            case 'B': typ[i] = 3; break;
        }
    }
    printf("%d\n", DFS(1, 0, 0, 0, 0));
}

然后博主就改写了一下变成了dp， 从大往小推（记忆化也是这样的）， 代码稍微复杂了一点：

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstdlib>
#define R register
#define IN inline
#define W while
#define MX 100050
char buf[MX];
int typ[MX], len, dp[4][4][4][4], res[4][4][4][4];//滚动消去一维
IN int get(const int &g1, const int &g2, const int &now)
{
    if(!g1 && !g2) return 1;
    if(!g1) return ((g2 != now) + 1);
    if(g1 == g2 && g2 == now) return 1;
    return (g1 != g2) + (g2 != now) + (g1 != now);
}
int main(void)
{
    scanf("%d%s", &len, buf + 1);
    for (R int i = 1; i <= len; ++i)
    {
        switch (buf[i])
        {
            case 'M': typ[i] = 1; break;
            case 'F': typ[i] = 2; break;
            case 'B': typ[i] = 3; break;
        }
    }
    R int a1, a2, b1, b2;
    for (R int i = len; i >= 1; --i)
    {
        for (a1 = 0; a1 <= 3; ++a1)
        for (a2 = 0; a2 <= 3; ++a2)
        for (b1 = 0; b1 <= 3; ++b1)
        for (b2 = 0; b2 <= 3; ++b2)
        {
            if((a1 && !a2) || (b1 && !b2)) continue;//非法状态需舍去
            res[a1][a2][b1][b2] = std::max(dp[a2][typ[i]][b1][b2] + get(a1, a2, typ[i]), dp[a1][a2][b2][typ[i]] + get(b1, b2, typ[i]));
        }
        std::swap(res, dp);
    }
    printf("%d\n", dp[0][0][0][0]);//初始状态为0，0，0，0
}