P1541 乌龟棋

Problem

传送门

P1541 乌龟棋

题目背景

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

题目描述

乌龟棋的棋盘是一行N个格子，每个格子上一个分数（非负整数）。棋盘第1格是唯一的起点，第N格是终点，游戏要求玩家控制一个乌龟棋子从起点出发走到终点。
乌龟棋中M张爬行卡片，分成4种不同的类型（MM张卡片中不一定包含所有4种类型的卡片，见样例），每种类型的卡片上分别标有1,2,3,4四个数字之一，表示使用这种卡片后，乌龟棋子将向前爬行相应的格子数。游戏中，玩家每次需要从所有的爬行卡片中选择一张之前没有使用过的爬行卡片，控制乌龟棋子前进相应的格子数，每张卡片只能使用一次。
游戏中，乌龟棋子自动获得起点格子的分数，并且在后续的爬行中每到达一个格子，就得到该格子相应的分数。玩家最终游戏得分就是乌龟棋子从起点到终点过程中到过的所有格子的分数总和。
很明显，用不同的爬行卡片使用顺序会使得最终游戏的得分不同，小明想要找到一种卡片使用顺序使得最终游戏得分最多。
现在，告诉你棋盘上每个格子的分数和所有的爬行卡片，你能告诉小明，他最多能得到多少分吗？

输入格式

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

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

第2行N个非负整数，a1,a2,…,aN其中 ai 表示棋盘第ii个格子上的分数。
第3行M个整数，b1,b2,…,bM表示M张爬行卡片上的数字。
输入数据保证到达终点时刚好用光M张爬行卡片。

输出格式

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

输入输出样例

输入

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

输出

73

说明/提示

• 每个测试点1s
• 小明使用爬行卡片顺序为1,1,3,1,2得到的分数为6+10+14+8+18+17=73。注意，由于起点是11，所以自动获得第1格的分数6
• 对于30%的数据有1≤N≤30,1≤M≤12
• 对于50%的数据有1≤N≤120,1≤M≤50，且4种爬行卡片，每种卡片的张数不会超过20
• 对于100%的数据有1≤N≤350,1≤M≤120，且4种爬行卡片，每种卡片的张数不会超过40,0≤ai≤100,1≤i≤N,1≤bi≤4,1≤i≤M

Solution

前言

这是蒟蒻 (我) 第一次写博客，写不好勿喷…

关于这道题

说实话，这道题这的很简单
这道题是一个简单的 DP，只是有四维,但我还是写了很久~

第一次

我一看，这不是爆搜吗？于是暴力一手，见ANS

第二次

我开始写DP，但是五维的~~（我也不知道为什么）~~，即用 f[i][j1][j2][j3][j4] 代表第i个用j1个1号，j2个2号，j3个3号，j4个4号的最优解，但空间爆了（洛谷评判机：No valid executable file was produced by the compiler）

第三次

写四维DP。用 f[j1][j2][j3][j4] 代表用j1个1号，j2个2号，j3个3号，j4个4号的最优解，顺利地完成了状态转移方程，打出了ANS

Ans

第一遍（爆搜）

只拿了 30分，其余TLE

#include<bits/stdc++.h>
#define FOR(ii,xx,yy) for(int ii=xx;ii<yy;ii++)
using namespace std;
int ans,n,m,cards[5],temp,a[400];
int dfs(int lef1,int lef2,int lef3,int lef4,int k,int point){
	if (point>ans)ans=point;
	if(lef1>0) dfs(lef1-1,lef2,lef3,lef4,k+1,point+a[k+1]);
	if(lef2>0) dfs(lef1,lef2-1,lef3,lef4,k+2,point+a[k+2]);
	if(lef3>0) dfs(lef1,lef2,lef3-1,lef4,k+3,point+a[k+3]);
	if(lef4>0) dfs(lef1,lef2,lef3,lef4-1,k+4,point+a[k+4]);
	return 0;
}
int main (){
	scanf("%d %d",&n,&m);
	FOR(i,0,n) scanf("%d",&a[i]);
	FOR(i,0,m) scanf("%d",&temp),cards[temp]++;
	dfs(cards[1],cards[2],cards[3],cards[4],0,a[0]);
	printf("%d",ans);
	return 0;
}

第二遍（五维！）

#include<bits/stdc++.h>
#define FOR(ii,xx,yy) for(register int ii=xx;ii<yy;ii++)
using namespace std;
int ans,n,m,cards[5],temp,a[400],f[360][45][45][45][45];

int main (){
	scanf("%d %d",&n,&m);
	FOR(i,0,n) scanf("%d",&a[i]);
	FOR(i,0,m) scanf("%d",&temp),cards[temp]++;
	f[0][cards[1]][cards[2]][cards[3]][cards[4]]=a[0];
	FOR(i,0,n){
		FOR(j1,0,cards[1]){
			FOR(j2,0,cards[1]){
				FOR(j3,0,cards[1]){
					FOR(j4,0,cards[1]){
						if(f[i][j1][j2][j3][j4]>0){
							if(j1>0) if(f[i][j1][j2][j3][j4]+a[i+1]>f[i+1][j1-1][j2  ][j3  ][j4  ])f[i+1][j1-1][j2][j3][j4]=f[i][j1][j2][j3][j4]+a[i+1];
							if(j2>0) if(f[i][j1][j2][j3][j4]+a[i+2]>f[i+2][j1  ][j2-1][j3  ][j4  ])f[i+2][j1  ][j2-1][j3  ][j4  ]=f[i][j1][j2][j3][j4]+a[i+2];
							if(j3>0) if(f[i][j1][j2][j3][j4]+a[i+3]>f[i+3][j1  ][j2  ][j3-1][j4  ])f[i+3][j1  ][j2  ][j3-1][j4  ]=f[i][j1][j2][j3][j4]+a[i+3];
							if(j4>0) if(f[i][j1][j2][j3][j4]+a[i+1]>f[i+4][j1  ][j2  ][j3  ][j4-1])f[i+4][j1  ][j2  ][j3  ][j4-1]=f[i][j1][j2][j3][j4]+a[i+4];
						}
					}
				}
			}
		}
	}
	printf("%d",f[n-1][0][0][0][0]);
	return 0;
}

第三遍（四维DP）

#include<bits/stdc++.h>
#define FOR(ii,xx,yy) for(register int ii=xx;ii<yy;ii++)
using namespace std;
int ans,n,m,cards[5],temp,a[400],f[41][41][41][41];

int main () {
	scanf("%d %d",&n,&m);
	FOR(i,0,n) scanf("%d",&a[i]);
	FOR(i,0,m) scanf("%d",&temp),cards[temp]++;
	f[0][0][0][0]=a[0];
	FOR(j1,0,cards[1]+1) {
		FOR(j2,0,cards[2]+1) {
			FOR(j3,0,cards[3]+1) {
				FOR(j4,0,cards[4]+1) {
					int i=j1+2*j2+3*j3+4*j4;
					if(j1<cards[1]) if(f[j1][j2][j3][j4]+a[i+1]>f[j1+1][j2  ][j3  ][j4  ])f[j1+1][j2  ][j3  ][j4  ]=f[j1][j2][j3][j4]+a[i+1];
					if(j2<cards[2]) if(f[j1][j2][j3][j4]+a[i+2]>f[j1  ][j2+1][j3  ][j4  ])f[j1  ][j2+1][j3  ][j4  ]=f[j1][j2][j3][j4]+a[i+2];
					if(j3<cards[3]) if(f[j1][j2][j3][j4]+a[i+3]>f[j1  ][j2  ][j3+1][j4  ])f[j1  ][j2  ][j3+1][j4  ]=f[j1][j2][j3][j4]+a[i+3];
					if(j4<cards[4]) if(f[j1][j2][j3][j4]+a[i+4]>f[j1  ][j2  ][j3  ][j4+1])f[j1  ][j2  ][j3  ][j4+1]=f[j1][j2][j3][j4]+a[i+4];
				}
			}
		}
	}
	printf("%d",f[cards[1]][cards[2]][cards[3]][cards[4]]);
	return 0;
}