【洛谷】P1541 乌龟棋

题目地址：

https://www.luogu.com.cn/problem/P1541

题目背景：
小明过生日的时候，爸爸送给他一副乌龟棋当作礼物。

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

输入格式：
每行中两个数之间用一个空格隔开。
第$1$行$2$个正整数$N,M$，分别表示棋盘格子数和爬行卡片数。
第$2$行$N$个非负整数，$a_1,a_2,\dots ,a_N$，其中$a_i$表示棋盘第$i$个格子上的分数。
第$3$行$M$个整数，$b_1,b_2,\dots ,b_M$，表示$M$张爬行卡片上的数字。输入数据保证到达终点时刚好用光$M$张爬行卡片。

输出格式：
$1$个整数，表示小明最多能得到的分数。

数据范围：
对于$30\%$的数据有$1\le N\le 30,1\le M\le 12$。
对于$50\%$的数据有$1\le N\le 120,1\le M\le 50$，且$4$种爬行卡片，每种卡片的张数不会超过$20$。
对于$100\%$的数据有$1\le N\le 350,1\le M\le 120$，且$4$种爬行卡片，每种卡片的张数不会超过$40$；$0\le a_i\le 100,1\le i\le N,1\le b_i\le 4,1\le i\le M$。

可以用记忆化搜索，设$f[x][b_1][b_2][b_3][b_4]$是当前位置是$x$，还剩$b_i$张数字$i$的卡片的情况下，能得到的最大分数；设$a[i]$是格子$i$上的分数。则有： f [ x ] [ b 1 ] [ b 2 ] [ b 3 ] [ b 4 ] = a [ x ] + max ⁡ { f [ x + 1 ] [ b 1 − 1 ] [ b 2 ] [ b 3 ] [ b 4 ] , f [ x + 2 ] [ b 1 ] [ b 2 − 1 ] [ b 3 ] [ b 4 ] , f [ x + 3 ] [ b 1 ] [ b 2 ] [ b 3 − 1 ] [ b 4 ] , f [ x + 4 ] [ b 1 ] [ b 2 ] [ b 3 ] [ b 4 − 1 ] } f[x][b_1][b_2][b_3][b_4]=a[x]+\max\{f[x+1][b_1-1][b_2][b_3][b_4], f[x+2][b_1][b_2-1][b_3][b_4], f[x+3][b_1][b_2][b_3-1][b_4],f[x+4][b_1][b_2][b_3][b_4-1]\} f[x][b1​][b2​][b3​][b4​]=a[x]+max{f[x+1][b1​−1][b2​][b3​][b4​],f[x+2][b1​][b2​−1][b3​][b4​],f[x+3][b1​][b2​][b3​−1][b4​],f[x+4][b1​][b2​][b3​][b4​−1]}代码如下：

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

const int N = 360, M = 50;
int n, m;
int a[N], b[5], f[M][M][M][M];

int dfs(int x, int b1, int b2, int b3, int b4) {
  int &v = f[b1][b2][b3][b4];
  if (~v) return v;
  if (!b1 && !b2 && !b3 && !b4) return v = a[x];
  if (b1) v = max(v, a[x] + dfs(x + 1, b1 - 1, b2, b3, b4));
  if (b2) v = max(v, a[x] + dfs(x + 2, b1, b2 - 1, b3, b4));
  if (b3) v = max(v, a[x] + dfs(x + 3, b1, b2, b3 - 1, b4));
  if (b4) v = max(v, a[x] + dfs(x + 4, b1, b2, b3, b4 - 1));
  return v;
}

int main() {
  scanf("%d%d", &n, &m);
  for (int i = 1; i <= n; i++) scanf("%d", &a[i]);
  for (int i = 1; i <= m; i++) {
    int x;
    b[scanf("%d", &x), x]++;
  }
  
  memset(f, -1, sizeof f);
  printf("%d\n", dfs(1, b[1], b[2], b[3], b[4]));
}

时空复杂度$O(b^4)$，$b$是每张牌的最大个数。