[NOIP2010 提高组] 乌龟棋

[NOIP2010 提高组] 乌龟棋

题目背景

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

题目描述

乌龟棋的棋盘是一行 $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$ 个整数，表示小明最多能得到的分数。

样例 #1

样例输入 #1

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

样例输出 #1

73

提示

每个测试点 1s。

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

对于 $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[c1][c2][c3][c4]表示用了4种类型的牌各使用了几张，搜索即可

代码

#include
#include
#include
#include
#include
#include
using namespace std;
const int N=1e3+23,MM=41;
int f[MM][MM][MM][MM];
int a[N],card[5];
int ans=0;
int n,m,tmp=1;
inline int read(int* x){
	scanf("%d",x);return *x;
}
int dfs(int cnt,int val,int c1,int c2,int c3,int c4){
	int &maxn=f[c1][c2][c3][c4];
	if(maxn) return maxn;
	if(cnt==n){
		return a[1];
	}
	if(c1+1<=card[1])
		maxn=max(maxn,dfs(cnt+1,val+a[cnt+1],c1+1,c2,c3,c4)+a[cnt+1]);
	if(c2+1<=card[2])
		maxn=max(maxn,dfs(cnt+2,val+a[cnt+2],c1,c2+1,c3,c4)+a[cnt+2]);
	if(c3+1<=card[3])
		maxn=max(maxn,dfs(cnt+3,val+a[cnt+3],c1,c2,c3+1,c4)+a[cnt+3]);
	if(c4+1<=card[4])
		maxn=max(maxn,dfs(cnt+4,val+a[cnt+4],c1,c2,c3,c4+1)+a[cnt+4]);
	return maxn;
}
signed main(){
// 	freopen("tortoise.in","r",stdin);
// 	freopen("tortoise.out","w",stdout);
	read(&n),read(&m);
	for(int i=1;i<=n;i++){
		read(&a[i]);
	}
	for(int i=1;i<=m;i++){
		int k;
		read(&k);
		card[k]++;
	}
	ans=dfs(1,a[1],0,0,0,0);
	cout<<ans<<endl;
// 	fclose(stdin);
// 	fclose(stdout);
	return 0;
}