洛谷 P1541 乌龟棋 —— DP

最新推荐文章于 2025-03-05 22:21:50 发布

转载最新推荐文章于 2025-03-05 22:21:50 发布 · 121 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Zinn/p/9757362.html

本文详细解析了洛谷P1541题目的动态规划解法，通过C++代码实现，展示了如何利用四维数组进行状态转移，最终求解出最优解。

题目：https://www.luogu.org/problemnew/show/P1541

DP。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int const xn=355,xm=45;
int n,m,a[xn],s[5],f[xm][xm][xm][xm];
int rd()
{
  int ret=0,f=1; char ch=getchar();
  while(ch<'0'||ch>'9'){if(ch=='-')f=0; ch=getchar();}
  while(ch>='0'&&ch<='9')ret=(ret<<3)+(ret<<1)+ch-'0',ch=getchar();
  return f?ret:-ret;
}
int main()
{
  n=rd(); m=rd();
  for(int i=1;i<=n;i++)a[i]=rd();
  for(int i=1,x;i<=m;i++)x=rd(),s[x]++;
  f[0][0][0][0]=a[1];
  for(int i=1;i<=m;i++)
    for(int j=0;j<=min(s[1],i);j++)
      for(int k=0;k<=min(s[2],i-j);k++)
        for(int l=0;l<=min(s[3],i-j-k);l++)
          {
              int t=i-j-k-l;
            f[j][k][l][t]=a[1+j+2*k+3*l+4*(i-j-k-l)];
            int ret=0;
            if(j)ret=max(ret,f[j-1][k][l][t]);
            if(k)ret=max(ret,f[j][k-1][l][t]);
            if(l)ret=max(ret,f[j][k][l-1][t]);
            if(i-j-k-l)ret=max(ret,f[j][k][l][t-1]);
            f[j][k][l][t]+=ret;
          }
  printf("%d\n",f[s[1]][s[2]][s[3]][s[4]]);
  return 0;
}