首页 > 代码库 > 洛谷 P1541 乌龟棋 Label:O(n^4)的dp
洛谷 P1541 乌龟棋 Label:O(n^4)的dp
题目背景
小明过生日的时候,爸爸送给他一副乌龟棋当作礼物。
题目描述
乌龟棋的棋盘是一行N个格子,每个格子上一个分数(非负整数)。棋盘第1格是唯一的起点,第N格是终点,游戏要求玩家控制一个乌龟棋子从起点出发走到终点。
乌龟棋中M张爬行卡片,分成4种不同的类型(M张卡片中不一定包含所有4种类型的卡片,见样例),每种类型的卡片上分别标有1、2、3、4四个数字之一,表示使用这种卡片后,乌龟棋子将向前爬行相应的格子数。游戏中,玩家每次需要从所有的爬行卡片中选择一张之前没有使用过的爬行卡片,控制乌龟棋子前进相应的格子数,每张卡片只能使用一次。
游戏中,乌龟棋子自动获得起点格子的分数,并且在后续的爬行中每到达一个格子,就得到该格子相应的分数。玩家最终游戏得分就是乌龟棋子从起点到终点过程中到过的所有格子的分数总和。
很明显,用不同的爬行卡片使用顺序会使得最终游戏的得分不同,小明想要找到一种卡片使用顺序使得最终游戏得分最多。
现在,告诉你棋盘上每个格子的分数和所有的爬行卡片,你能告诉小明,他最多能得到多少分吗?
输入输出格式
输入格式:
输入文件的每行中两个数之间用一个空格隔开。
第1行2个正整数N和M,分别表示棋盘格子数和爬行卡片数。
第2行N个非负整数,a1a2……aN,其中ai表示棋盘第i个格子上的分数。
第3行M个整数,b1b2……bM,表示M张爬行卡片上的数字。
输入数据保证到达终点时刚好用光M张爬行卡片。
输出格式:
输出只有1行,1个整数,表示小明最多能得到的分数。
输入输出样例
9 56 10 14 2 8 8 18 5 171 3 1 2 1
73
说明
每个测试点1s
小明使用爬行卡片顺序为1,1,3,1,2,得到的分数为6+10+14+8+18+17=73。注意,由于起点是1,所以自动获得第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。
代码
1 #include<iostream> 2 #include<cstdio> 3 #include<cstring> 4 #include<algorithm> 5 #define inf 0x3f3f3f3f 6 using namespace std; 7 int f[50][50][50][50],b[50],a[5000],N,M; 8 int main(){ 9 // freopen("01.txt","r",stdin);10 scanf("%d%d",&N,&M);11 12 for(int i=1;i<=N;i++)13 scanf("%d",&a[i]);14 15 for(int i=1;i<=M;i++){16 int x;17 scanf("%d",&x);18 ++b[x];19 }20 21 for(int i=0;i<=b[1];i++){22 for(int j=0;j<=b[2];j++){23 for(int k=0;k<=b[3];k++){24 for(int l=0;l<=b[4];l++){25 int mm=-inf;26 if(i>0)27 mm=max(mm,f[i-1][j][k][l]);28 if(j>0)29 mm=max(mm,f[i][j-1][k][l]);30 if(k>0)31 mm=max(mm,f[i][j][k-1][l]);32 if(l>0)33 mm=max(mm,f[i][j][k][l-1]);34 f[i][j][k][l]=mm+a[1+i*1+j*2+k*3+l*4];35 if(i==0&&j==0&&k==0&&l==0) f[i][j][k][l]+=inf;36 }37 }38 }39 }40 41 printf("%d\n",f[b[1]][b[2]][b[3]][b[4]]);42 return 0;43 }嗯,O(n^4)
直接转载:
用一个数组记录每种卡片出现的张数,用num数组记录,用a数组记录每个位置的分数。很明显,这是一道动态规划的题,分析一下,我们发现,可以不考虑卡片的使用顺序,而直接考虑使用的张数(想想为什么)。所以每一个状态我们只涉及此时的卡片的张数,用一个数组c[i][j][k][t]表示每种卡片的使用张数,容易得出状态转移方程
f[i][j][k][t]=max{f[i-1][j][k][t],f[i][j-1][k][t],f[i][j][k-1][t],f[i][j][k][t-1]}+ai+j*2+k*3+t*4+1
最后,题目上说恰好用完,那么就输出f[b[1]][b[2]][b[3]][b[4]]就可以了。需要注意的是,这是一个四重循环,且每个循环变量都是从0开始的,那么减去1之后会超界,所以需要注意一下,判断是否为0就行了。
洛谷 P1541 乌龟棋 Label:O(n^4)的dp