洛谷刷题C++语言 | P1541 乌龟棋

学习C++从娃娃抓起！记录下洛谷C++学习和备考过程中的题目，记录每一个瞬间。

附上汇总贴：洛谷刷题C++语言 | 汇总

---

【题目描述】

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

乌龟棋的棋盘是一行 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 张爬行卡片。

【输出】

一个整数，表示小明最多能得到的分数。

【输入样例】

9 5
6 10 14 2 8 8 18 5 17
1 3 1 2 1

【输出样例】

73

【代码详解】

// 使用记忆化DFS
#include <bits/stdc++.h>
using namespace std;
int x, n, m;
int a[355], b[5], dp[45][45][45][45];
int dfs(int pos)
{
    int s = 0;
    if (pos==n) return a[n];
    if (dp[b[1]][b[2]][b[3]][b[4]]) return dp[b[1]][b[2]][b[3]][b[4]];
    for (int i=1; i<=4; i++) {
        if (b[i]==0) continue;
        b[i]--;
        s = max(s, dfs(pos+i));
        b[i]++;
    }
    return dp[b[1]][b[2]][b[3]][b[4]] = s+a[pos];
}
int main()
{
    cin >> n >> m;
    for (int i=1; i<=n; i++) cin >> a[i];
    for (int i=1; i<=m; i++) {
        cin >> x;
        b[x]++;
    }
    cout << dfs(1);
    return 0;
}

// 动态规划
#include <bits/stdc++.h>
using namespace std;
int a[355], dp[45][45][45][45], b[5];
int n, m;
int main()
{
    cin >> n >> m;
    for (int i=1; i<=n; i++)
        cin >> a[i];
    for (int i=1; i<=m; i++) {
        int x; cin >> x;
        b[x]++;
    }
    dp[0][0][0][0] = a[1];
    for (int x1=0; x1<=b[1]; x1++) {
        for (int x2=0; x2<=b[2]; x2++) {
            for (int x3=0; x3<=b[3]; x3++) {
                for (int x4=0; x4<=b[4]; x4++) {
                    int z = a[1+x1+x2*2+x3*3+x4*4];
                    if (x1!=0) dp[x1][x2][x3][x4] = max(dp[x1][x2][x3][x4], dp[x1-1][x2][x3][x4]+z);
                    if (x2!=0) dp[x1][x2][x3][x4] = max(dp[x1][x2][x3][x4], dp[x1][x2-1][x3][x4]+z);
                    if (x3!=0) dp[x1][x2][x3][x4] = max(dp[x1][x2][x3][x4], dp[x1][x2][x3-1][x4]+z);
                    if (x4!=0) dp[x1][x2][x3][x4] = max(dp[x1][x2][x3][x4], dp[x1][x2][x3][x4-1]+z);
                }
            }
        }
    }
    cout << dp[b[1]][b[2]][b[3]][b[4]];
    return 0;
}

【运行结果】

9 5
6 10 14 2 8 8 18 5 17
1 3 1 2 1
73