洛谷 1541 乌龟棋——dp

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

以用了几张牌为阶段。注意知道了用了4种牌各几张后，当前位置就是确定的，所以不用记录什么的。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=45,M=355;
int n,m,lm[5],a[M],dp[N][N][N][N],ans;
int rdn()
{
  int ret=0;bool fx=1;char ch=getchar();
  while(ch>'9'||ch<'0'){if(ch=='-')fx=0;ch=getchar();}
  while(ch>='0'&&ch<='9') ret=(ret<<3)+(ret<<1)+ch-'0',ch=getchar();
  return fx?ret:-ret;
}
int main()
{
  n=rdn(); m=rdn();
  for(int i=0;i<n;i++) a[i]=rdn();
  n--;
  for(int i=1,d;i<=m;i++)
    {
      d=rdn(); lm[d]++;
    }
  dp[0][0][0][0]=a[0];
  for(int t=1;t<=m;t++)
    for(int i=0;i<=lm[1];i++)
      for(int j=0;j<=lm[2]&&j<=t-i;j++)
    for(int k=0;k<=lm[3]&&k<=t-i-j;k++)
      {
        int l=t-i-j-k,d=i+(j<<1)+k*3+(l<<2);
        if(d>n)continue;
        int ret=0;
        if(i)ret=max(ret,dp[i-1][j][k][l]);
        if(j)ret=max(ret,dp[i][j-1][k][l]);
        if(k)ret=max(ret,dp[i][j][k-1][l]);
        if(l)ret=max(ret,dp[i][j][k][l-1]);
        dp[i][j][k][l]=ret+a[d];
        if(d==n)ans=max(ans,dp[i][j][k][l]);
      }
  printf("%d\n",ans);
  return 0;
}