wikioi-天梯-提高一等-背包dp-1068：乌龟棋

题目描述 Description

小明过生日的时候，爸爸送给他一副乌龟棋当作礼物。 乌龟棋的棋盘是一行N个格子，每个格子上一个分数（非负整数）。棋盘第1格是唯一 的起点，第N格是终点，游戏要求玩家控制一个乌龟棋子从起点出发走到终点。

…… 1 2 3 4 5 ……N 乌龟棋中M张爬行卡片，分成4种不同的类型（M张卡片中不一定包含所有4种类型 的卡片，见样例），每种类型的卡片上分别标有1、2、3、4四个数字之一，表示使用这种卡 片后，乌龟棋子将向前爬行相应的格子数。游戏中，玩家每次需要从所有的爬行卡片中选择 一张之前没有使用过的爬行卡片，控制乌龟棋子前进相应的格子数，每张卡片只能使用一次。 游戏中，乌龟棋子自动获得起点格子的分数，并且在后续的爬行中每到达一个格子，就得到 该格子相应的分数。玩家最终游戏得分就是乌龟棋子从起点到终点过程中到过的所有格子的 分数总和。 很明显，用不同的爬行卡片使用顺序会使得最终游戏的得分不同，小明想要找到一种卡 片使用顺序使得最终游戏得分最多。 现在，告诉你棋盘上每个格子的分数和所有的爬行卡片，你能告诉小明，他最多能得到 多少分吗？

输入描述 Input Description

输入的每行中两个数之间用一个空格隔开。 第1行2个正整数N和M，分别表示棋盘格子数和爬行卡片数。 第2行N个非负整数，a1a2……aN

，其中ai表示棋盘第i个格子上的分数。 第3行M个整数，b1b2……bM

，表示M张爬行卡片上的数字。 输入数据保证到达终点时刚好用光M张爬行卡片，即N - 1=∑(1->M) b_i

输出描述 Output Description

输出一行一个整数

样例输入 Sample Input

13 8

4 96 10 64 55 13 94 53 5 24 89 8 30

1 1 1 1 1 2 4 1

样例输出 Sample Output

455

数据范围及提示 Data Size & Hint

【数据范围】

对于30%的数据有1 ≤ N≤ 30，1 ≤M≤ 12。

对于50%的数据有1 ≤ N≤ 120，1 ≤M≤ 50，且4 种爬行卡片，每种卡片的张数不会超

过20。

~

对于100%的数据有1 ≤ N≤ 350，1 ≤M≤ 120，且4 种爬行卡片，每种卡片的张数不会

超过40；0 ≤ ai ≤ 100，1 ≤ i ≤ N；1 ≤ bi ≤ 4，1 ≤ i ≤M。输入数据保证N−1=Σi b

类型：dp  难度：2

题意：一条路长为N，给N个数，表示路上每格的分数。要从第1格走到第N格，给M个卡片，卡片上的数字和为N-1，每次从卡片中拿一张，走相应的步数，只能往前走，卡片的值为1-4，走到相应的格子时获得相应的分数，问按照什么顺序走，获取的分数最高。

分析：稍微有点绕的dp。本来考虑用dp[i]表示走i步所获得的的最高分数，但是发现结果不对，细想发现dp[i]无法表示一个状态，比如有三张卡片1,1,2，那么dp[2]既能表示走了一个2，也能表示走了两个1，而且这两种走法获取的分数不同，所以光靠步数不足以表示一个状态，还需要记录目前能使用的牌，或已经用过的牌，才能确定一个状态。

于是，状态变成：dp[i][a][b][c][d]，表示用a个1，b个2，c个3，d个4这些牌，走i步（还需要走的步数），所获得的最高分（不包括当前格子的分数），设s[i]为第i个格子的分数

状态转移方程：dp[i][a][b][c][d] = max(dp[i-1][a-1][b][c][d]+s[i-1], dp[i-2][a][b-1][c][d]+s[i-2], dp[i-3][a][b][c-1][d]+s[i-3], dp[i-4][a][b][c][d-1]+s[i-4])

即每一步都选择4中走法，选得分最高的那个

优化：发现数组占用内存过大，需要降维，发现去掉i,a,b,c,d其中一个仍然能够计算出另一个，因为i = a+2b+3c+4d，所以可以去掉维数最大的i维，节省内存。

代码：

 

#include<iostream>
#include<cstring>
using namespace std;

int mymax(int a,int b)
{
    return a>b?a:b;
}

short f[150],dp[40][40][40][40],s[400],n,m;

short fun(short a,short b,short c,short d)
{
    short left = a+2*b+3*c+4*d;
    if(!left) return 0;
    if(dp[a][b][c][d]>0) return dp[a][b][c][d];
    
    if(a>0)
    {
        dp[a][b][c][d]= mymax(dp[a][b][c][d],
           fun(a-1,b,c,d)+s[left-1]);
    }
    if(b>0)
    {
        dp[a][b][c][d] = mymax(dp[a][b][c][d],
           fun(a,b-1,c,d)+s[left-2]);
    }
    if(c>0)
    {
        dp[a][b][c][d] = mymax(dp[a][b][c][d],
           fun(a,b,c-1,d)+s[left-3]);
    }
    if(d>0)
    {
        dp[a][b][c][d] = mymax(dp[a][b][c][d],
           fun(a,b,c,d-1)+s[left-4]);
    }
    
    return dp[a][b][c][d];
}

int main()
{
    short cnt[4];
    memset(cnt,0,sizeof(cnt));
    cin>>n>>m;
    for(int i=0; i<n; i++)
        cin>>s[i];
    for(int i=0; i<m; i++)
    {
        int t;
        cin>>t;
        cnt[t-1]++;
    }
    memset(f,0,sizeof(f));
    memset(dp,0,sizeof(dp));
    
    cout<<fun(cnt[0],cnt[1],cnt[2],cnt[3])+s[n-1]<<endl;
}