小明过生日的时候,爸爸送给他一副乌龟棋当作礼物。
乌龟棋的棋盘只有一行,该行有 N N N 个格子,每个格子上一个分数(非负整数)。
棋盘第 1 1 1 格是唯一的起点,第 N N N 格是终点,游戏要求玩家控制一个乌龟棋子从起点出发走到终点。
乌龟棋中共有 M M M 张爬行卡片,分成 4 4 4 种不同的类型( M M M 张卡片中不一定包含所有 4 4 4 种类型的卡片),每种类型的卡片上分别标有 1 、 2 、 3 、 4 1、2、3、4 1、2、3、4 四个数字之一,表示使用这种卡片后,乌龟棋子将向前爬行相应的格子数。
游戏中,玩家每次需要从所有的爬行卡片中选择一张之前没有使用过的爬行卡片,控制乌龟棋子前进相应的格子数,每张卡片只能使用一次。
游戏中,乌龟棋子自动获得起点格子的分数,并且在后续的爬行中每到达一个格子,就得到该格子相应的分数。
玩家最终游戏得分就是乌龟棋子从起点到终点过程中到过的所有格子的分数总和。
很明显,用不同的爬行卡片使用顺序会使得最终游戏的得分不同,小明想要找到一种卡片使用顺序使得最终游戏得分最多。
现在,告诉你棋盘上每个格子的分数和所有的爬行卡片,你能告诉小明,他最多能得到多少分吗?
输入格式
输入文件的每行中两个数之间用一个空格隔开。
第 1 1 1 行 2 2 2 个正整数 N N N 和 M M M,分别表示棋盘格子数和爬行卡片数。
第 2 2 2 行 N N N 个非负整数, a 1 , a 2 , … … , a N a_1,a_2,……,a_N a1,a2,……,aN,其中 a i a_i ai 表示棋盘第 i i i 个格子上的分数。
第 3 3 3 行 M M M 个整数, b 1 , b 2 , … … , b M b_1,b_2,……,b_M b1,b2,……,bM,表示 M M M 张爬行卡片上的数字。
输入数据保证到达终点时刚好用光 M M M 张爬行卡片。
输出格式
输出只有
1
1
1 行,包含
1
1
1 个整数,表示小明最多能得到的分数。
数据范围
1
≤
N
≤
350
,
1
≤
M
≤
120
,
0
≤
a
i
≤
100
,
1
≤
b
i
≤
4
1≤N≤350,\\ 1≤M≤120,\\ 0≤a_i≤100,\\ 1≤b_i≤4
1≤N≤350,1≤M≤120,0≤ai≤100,1≤bi≤4
每种爬行卡片的张数不会超过
40
40
40。
输入样例:
9 5
6 10 14 2 8 8 18 5 17
1 3 1 2 1
输出样例:
73
- 思路 (线性DP)
O
(
4
0
4
)
O(40^4)
O(404)
状态表示: f [ b 1 , b 2 , b 3 , b 4 ] f[b_1,b_2,b_3,b_4] f[b1,b2,b3,b4] 表示所有第 i i i 种卡片使用了 b i b_i bi 张的走法的最大分值。
状态计算:将 f [ b 1 , b 2 , b 3 , b 4 ] f[b_1,b_2,b_3,b_4] f[b1,b2,b3,b4]表示的所有走法按最后一步选择哪张卡片分成四类:第 i i i 类为最后一步选择第 i i i 种卡片。比如 i = 2 i=2 i=2,则这一类的最大分值是 f [ b 1 , b 2 − 1 , b 3 , b 4 ] + s c o r e [ b 1 + 2 b 2 + 3 b 3 + 4 b 4 ] f[b_1,b_2−1,b_3,b_4]+score[b_1+2b_2+3b_3+4b_4] f[b1,b2−1,b3,b4]+score[b1+2b2+3b3+4b4]
时间复杂度
一共有 4 0 4 40^4 404 个状态,每个状态需要 O ( 1 ) O(1) O(1) 的计算量,因此总时间复杂度是 O ( 4 0 4 ) O(40^4) O(404)。
#include<bits/stdc++.h>
using namespace std;
const int maxn=410;
int n;
int a[maxn];
int f[45][45][45][45];
int b[5];
int m;
int main() {
scanf("%d%d", &n, &m);
for (int i = 0; i < n; i++) {
scanf("%d", &a[i]);
}
for (int i = 1, x; i <= m; i++) {
scanf("%d", &x);
b[x]++;
}
for (int i = 0; i <= b[1]; i++) {
for (int j = 0; j <= b[2]; j++) {
for (int k = 0; k <= b[3]; k++) {
for (int l = 0; l <= b[4]; l++) {
int x = a[i + j * 2 + k * 3 + l * 4];
int &mx = f[i][j][k][l];
mx = x;
if (i) mx = max(mx, f[i - 1][j][k][l] + x);
if (j) mx = max(mx, f[i][j - 1][k][l] + x);
if (k) mx = max(mx, f[i][j][k - 1][l] + x);
if (l) mx = max(mx, f[i][j][k][l - 1] + x);
}
}
}
}
printf("%d\n", f[b[1]][b[2]][b[3]][b[4]]);
return 0;
}