【动态规划】 乌龟棋

354 阅读4分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第5天,点击查看活动详情

[NOIP2010 提高组] 乌龟棋

题目背景

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

题目描述

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

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

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

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

现在,告诉你棋盘上每个格子的分数和所有的爬行卡片,你能告诉小明,他最多能得到多少分吗?

输入格式

每行中两个数之间用一个空格隔开。

1122个正整数N,MN,M,分别表示棋盘格子数和爬行卡片数。

22NN个非负整数,a1,a2,,aNa_1,a_2,…,a_N,其中aia_i表示棋盘第ii个格子上的分数。

33MM个整数,b1,b2,,bMb_1,b_2,…,b_M,表示M张爬行卡片上的数字。

输入数据保证到达终点时刚好用光MM张爬行卡片。

输出格式

11个整数,表示小明最多能得到的分数。

样例 #1

样例输入 #1

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

样例输出 #1

73

提示

每个测试点1s1s

小明使用爬行卡片顺序为1,1,3,1,21,1,3,1,2,得到的分数为6+10+14+8+18+17=736+10+14+8+18+17=73。注意,由于起点是11,所以自动获得第11格的分数66

对于30%30\%的数据有1N30,1M121≤N≤30,1≤M≤12

对于50%50\%的数据有1N120,1M501≤N≤120,1≤M≤50,且44种爬行卡片,每种卡片的张数不会超过2020

对于100%100\%的数据有1N350,1M1201≤N≤350,1≤M≤120,且44种爬行卡片,每种卡片的张数不会超过40400ai100,1iN,1bi4,1iM0≤a_i≤100,1≤i≤N,1≤b_i≤4,1≤i≤M

解题思路

一共有四张卡片,代表所能走的不同的步数,也就是当前数的取得,有可能是在前一个数的基础上,走了一步/两步/三步/四步而来,因而考虑用一个四维的状态数组f[i][j][k][h]分别表示卡片的张数,DP列出所有可能的状态即可。

要注意的细节:

  1. 第一格是唯一的起点,所以f[0][0][0][0]=a[1],也就是一张卡片都没有用的时候,取得的初值。
  2. 如果用了第一张卡片,则用f[1][0][0][0]表示,有f[1][0][0][0]=f[0][0][0][0]+a[2],也就是从第一格开始,走了一格,实际上要取的数是a[2],也就是a[t+1],其中t为卡片所规定的步数。
  3. 每张卡片只能用一次,但每一种卡片不只一张,因此在输入的时候,需要做个统计。
  4. 注意动规的初值,循环要从0开始,表示没有用到相应的卡片。
#include<cstdio>
#include<algorithm>
#define N 360
#define M 130
using namespace std;
int n,m,x,a[N],b[5];
int f[M][M][M][M];
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++)
		{
			scanf("%d",&x);
			b[x]++;
		}
		f[0][0][0][0]=a[1];
		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 h=0;h<=b[4];h++)
		{
			x=i+j*2+k*3+h*4;
			if(i) f[i][j][k][h]=max(f[i][j][k][h],f[i-1][j][k][h]+a[x+1]);
			if(j) f[i][j][k][h]=max(f[i][j][k][h],f[i][j-1][k][h]+a[x+1]);
			if(k) f[i][j][k][h]=max(f[i][j][k][h],f[i][j][k-1][h]+a[x+1]);
			if(h) f[i][j][k][h]=max(f[i][j][k][h],f[i][j][k][h-1]+a[x+1]);
		}
		printf("%d\n",f[b[1]][b[2]][b[3]][b[4]]);
		return 0;
}