题目: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 ",ans); return 0; }