【动态规划、dp】P1541 [NOIP2010 提高组] 乌龟棋 题解

题目描述

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

乌龟棋的棋盘是一行 $N$ 个格子，每个格子上一个分数（非负整数）。棋盘第 $1$ 格是唯一的起点，第 $N$ 格是终点，游戏要求玩家控制一个乌龟棋子从起点出发走到终点。

乌龟棋中 $M$ 张爬行卡片，分成 $4$ 种不同的类型（$M$ 张卡片中不一定包含所有 $4$ 种类型的卡片，见样例），每种类型的卡片上分别标有 $1,2,3,4$ 四个数字之一，表示使用这种卡片后，乌龟棋子将向前爬行相应的格子数。游戏中，玩家每次需要从所有的爬行卡片中选择一张之前没有使用过的爬行卡片，控制乌龟棋子前进相应的格子数，每张卡片只能使用一次。

游戏中，乌龟棋子自动获得起点格子的分数，并且在后续的爬行中每到达一个格子，就得到该格子相应的分数。玩家最终游戏得分就是乌龟棋子从起点到终点过程中到过的所有格子的分数总和。

很明显，用不同的爬行卡片使用顺序会使得最终游戏的得分不同，小明想要找到一种卡片使用顺序使得最终游戏得分最多。

现在，告诉你棋盘上每个格子的分数和所有的爬行卡片，你能告诉小明，他最多能得到多少分吗？输入数据保证到达终点时刚好用光 $M$ 张爬行卡片。

对于 $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$。

思路

令 $d{p}_{i,j,k,l}$ 表示使用 $i$ 张第 $1$ 种卡片，$j$ 张第 $2$ 种卡片,$k$ 张第 $3$ 种卡片，$l$ 张第 $4$ 种卡片的最大得分，题目要求输出的即为 $d{p}_{l1,l2,l3,l4}$，$l1,l2,l3,l4$ 分别表示四种卡片数量。

边界条件：$d{p}_{0,0,0,0}=a_1$，显而易见乌龟棋子自动获得起点格子（ 就是不经过任何操作）的分数。

转移方程：容易发现：$d{p}_{i,j,k,l}=\max (d{p}_{i-1,j,k,l},d{p}_{i,j-1,k,l},d{p}_{i,j,k-1,l},d{p}_{i,j,k,l-1})+a_{i+j\times 2+k\times 3+l\times 4+1}$，至于 $a_{i+j\times 2+k\times 3+l\times 4+1}$ 而不是 $a_{i+j\times 2+k\times 3+l\times 4}$，是因为我们保存时起始点是 $a_1$。

时间复杂度 $O(l{1}^4)$ ，空间可采用滚动数组优化成三次方的，但是我懒 。

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
int dp[41][41][41][41];
int n,m;
int a[355];
int l1,l2,l3,l4;
signed main() {
	scanf("%lld %lld",&n,&m);
	for(int i = 1;i <= n;i++) scanf("%lld",&a[i]);
	while(m--) {
		int inp;
		scanf("%lld",&inp);
		if(inp == 1) l1++;
		else if(inp == 2) l2++;
		else if(inp == 3) l3++;
		else l4++;
	}
	dp[0][0][0][0] = a[1];
	for(int i = 0;i <= l1;i++) {
		for(int j = 0;j <= l2;j++) {
			for(int k = 0;k <= l3;k++) {
				for(int l = 0;l <= l4;l++) {
					if(i + j + k + l == 0) continue;
					if(i != 0) dp[i][j][k][l] = max(dp[i][j][k][l],dp[i - 1][j][k][l]);
					if(j != 0) dp[i][j][k][l] = max(dp[i][j][k][l],dp[i][j - 1][k][l]);
					if(k != 0) dp[i][j][k][l] = max(dp[i][j][k][l],dp[i][j][k - 1][l]);
					if(l != 0) dp[i][j][k][l] = max(dp[i][j][k][l],dp[i][j][k][l - 1]);
					dp[i][j][k][l] += a[i + j * 2 + k * 3 + l * 4 + 1];
				}
			}
		}
	}
	printf("%lld\n",dp[l1][l2][l3][l4]);
    return 0;
}