这题一眼DP,本来是设五维的:
f[i][j][k][l][o]表示到了第i格还剩下j个‘1’,k个‘2’,l个‘3’,o个‘4’
但是,咳咳,空超了!!!
如何?我们在仔细地看一看DP数组
我们可以发现,
i其实是不需要的!
为什么?!
因为,当我们确定了剩下的卡片,我们就能确定了它当前所在的位置是哪里。
所以,我们可以直接将i去掉,perfect!
那这样的话,我们只要设四维就可以了。
f[j][k][l][o]表示还剩下j个‘1’,k个‘2’,l个‘3’,o个‘4’
ok,这样大家都听懂了吧~
上标:
#include<cstdio>
#define max(x,y) x=x<y ? y:x
using namespace std;
int f[41][41][41][41];
int n,m,a[351],b[5];
inline int read()
{
int x=0; char c=getchar();
while (c<'0' || c>'9') c=getchar();
while (c>='0' && c<='9') x=(x<<1)+(x<<3)+(c^48),c=getchar();
return x;
}
int main()
{
// ke = king eight = wang ba = wu'gui
// freopen("ke.in","r",stdin);
// freopen("ke.out","w",stdout);
n=read(),m=read();
for (int i=1;i<=n;i++) a[i]=read();
for (int i=1;i<=m;i++) b[read()]++;
f[b[1]][b[2]][b[3]][b[4]]=a[1];
for (int i=2;i<=n;i++)
for (int j=0;j<=b[1];j++)
for (int k=0;k<=b[2];k++)
for (int l=0;l<=b[3];l++)
for (int o=0;o<=b[4];o++)
if (j+2*k+3*l+4*o==n-i)
{
if (j<b[1]) max(f[j][k][l][o],f[j+1][k][l][o]+a[i]);
if (k<b[2] && i>1) max(f[j][k][l][o],f[j][k+1][l][o]+a[i]);
if (l<b[3] && i>2) max(f[j][k][l][o],f[j][k][l+1][o]+a[i]);
if (o<b[4] && i>3) max(f[j][k][l][o],f[j][k][l][o+1]+a[i]);
// printf("%d %d %d %d %d\n",j,k,l,o,f[j][k][l][o]);
}
printf("%d\n",f[0][0][0][0]);
return 0;
}