bzoj1806&&vijos p1386(递推)

链接：点击打开链接

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

代码：

#include <set>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
char s[100005];
int dp[2][4][4][4][4];
int change(char c){
    if(c=='M')
    return 1;
    if(c=='F')
    return 2;
    return 3;
}
int cal(int a,int b,int c){                     //判断三个有几种
    set<int> s;
    if(a!=0)
    s.insert(a);
    if(b!=0)
    s.insert(b);
    if(c!=0)
    s.insert(c);
    return s.size();
}   
int main(){                                     //dp[i][a][b][c][d]表示到第i个字母，第一个矿
    int n,i,j,a,b,c,d,ans,tmp,tmp_c;            //最后两种是a,b，第二个矿最后两个是c,d的最大值
    while(scanf("%d",&n)!=EOF){
        scanf("%s",s+1);
        memset(dp,-1,sizeof(dp));
        dp[0][0][0][0][0]=0;
        for(i=1;i<=n;i++){
            memset(dp[i%2],-1,sizeof(dp[i%2]));
            for(a=0;a<4;a++){                   //用滚动数组
                for(b=0;b<4;b++){       
                    for(c=0;c<4;c++){
                        for(d=0;d<4;d++){
                            if(dp[(i-1)%2][a][b][c][d]==-1)
                            continue;
                            tmp_c=change(s[i]);
                            tmp=cal(a,b,tmp_c); //转移的时候直接往后推，最后两个变成前两个
                            dp[i%2][b][tmp_c][c][d]=max(dp[i%2][b][tmp_c][c][d],dp[(i-1)%2][a][b][c][d]+tmp);
                            tmp=cal(c,d,tmp_c);
                            dp[i%2][a][b][d][tmp_c]=max(dp[i%2][a][b][d][tmp_c],dp[(i-1)%2][a][b][c][d]+tmp);
                        }
                    }
                }
            }
        }
        ans=0;
        for(a=0;a<4;a++)
        for(b=0;b<4;b++)
        for(c=0;c<4;c++)
        for(d=0;d<4;d++)
        ans=max(ans,dp[n%2][a][b][c][d]);
        printf("%d\n",ans);
    }
    return 0;
}